Fixing bug :(