small fix in test