Trying fixing the test