Got rid of the unnecessary local variable
Corrected assert_equal's usage