making it work with tests