Add missing imports in subscription