Used assert instead of assert_equal