Trying to fix a test