Fix the test names to match the actual assertion