small fix in test class