Updated small bugfix