Fix model test