Fix annotation

Fix return parameter. 

*Sorry for this.*