small fix for tests