Fix inaccurate return type in comment