Stage should be imported when it's used

- although it works even without it (by accident)