Fix tests to actually test for equality