Fixing build for not checking migration