Fix stage module name