quick fix in the tests