Fixed a test failure that was caused by the model change.