fix problem with tests