Changed assert => assert_equal