Fix to stage size