small bugfix after testing