Correcting method return type