Fix new model in tests