Move resize code to stage, instead of frame