Swap out assert for assert_equal