Fix annotation

Annotation mistakenly listed @return as a @param