fix stage size