small fix in testing