Use assert_equal where possible