Add missing Query model migration