Try to fix the test