Fix stage support
Merge pull request #4 from pkgdemon/patch-4

Fix stage support