fix stage map