Added small bugfix