Should fix the test that were failing