Bug fix

Bug fix