Add missing import 'core.models.query'