Bug fix

Fixed this bug