Fixing doc comment