Fix sw-stage annotation error
Bump version v0.5.1
Merge branch 'release/v0.5.1'