Got rid of the unnecessary local variable
Corrected model_test test cases