should be a better way to do Stage