Use subscription() as a decorator.
Fixing imports in tests.