[fix] - tests work properly