Add a hack to be able to run the tests