Fixed tests that had incorrect assert