Fix the last test with a different approach