fixed issue with tests