Test fix

small change