Add missing import in subscription handler