Fix the use of an unimported get_models