Fixed model test bug