diff --git a/urfave/semver.go b/urfave/semver.go
index f831e0f..21a1632 100644
--- a/urfave/semver.go
+++ b/urfave/semver.go
@@ -72,7 +72,7 @@ func semVerFlags() []cli.Flag {
 	}
 }
 
-// semVerFromContext creates a drone.Step from the cli.Context.
+// semVerFromContext creates a drone.SemVer from the cli.Context.
 func semVerFromContext(ctx *cli.Context) drone.SemVer {
 	return drone.SemVer{
 		Version:    ctx.String("semver.version"),