Fix assertion

assert_equal should be used here, not assert