Small fix in update 1.1