Correcting return type