bug fix

Fixing test