Fix for the test?