Trying to fix a test case...