Simple test fix

simple fix