Apply suggestions from code review
Apply suggestions from code review

Thanks for the suggestions @dbrattli, it is a cleaner solution.

Co-authored-by: Dag Brattli <e134adc54f5fadca0d923da7a0a979d532364523@brattli.net>
dbrattli/aioreactive