Fixing bug in last update