Bug fixed

Fixed first bug