refactored
Small bug fix