Fixing bug

Уииии! \o/