Add missing import in subscription