test fix

small fix