diff --git a/cmd/prune.go b/cmd/prune.go
index 0004ac3..cf5ad14 100644
--- a/cmd/prune.go
+++ b/cmd/prune.go
@@ -44,7 +44,7 @@ var pruneCmd = &cobra.Command{
 
 // init initializes prune command.
 func init() {
-	flags := rmCmd.Flags()
+	flags := pruneCmd.Flags()
 	flags.BoolVar(&pruneConfig.DryRun, "dry-run", false, "do not remove any blobs, just print what would be removed")
 	flags.BoolVar(&pruneConfig.RemoveUntagged, "remove-untagged", true, "remove untagged manifests")
 
diff --git a/cmd/version.go b/cmd/version.go
index a7c39a5..d066048 100644
--- a/cmd/version.go
+++ b/cmd/version.go
@@ -39,7 +39,7 @@ var versionCmd = &cobra.Command{
 
 // init initializes version command.
 func init() {
-	flags := rmCmd.Flags()
+	flags := versionCmd.Flags()
 
 	if err := viper.BindPFlags(flags); err != nil {
 		panic(fmt.Errorf("bind version flags to viper: %w", err))