Fix return in define_method