Try to fix failure after test