Fix a bug in the model specs