Another small bugfix