Rewrote the model test class.