- fix bug with tests;