fixed an issue with the tests