Fixing bug

Fixed :)