small fix in tests