Fix return to warn