bug fix

fixed bug after running unit tests