Fix on the tests