fix unitests