diff --git a/lib/puppet/parser/functions/docker_plugin_enable_flags.rb b/lib/puppet/parser/functions/docker_plugin_enable_flags.rb
index 409eff5..ad53254 100644
--- a/lib/puppet/parser/functions/docker_plugin_enable_flags.rb
+++ b/lib/puppet/parser/functions/docker_plugin_enable_flags.rb
@@ -1,16 +1,20 @@
 # frozen_string_literal: true
 
 require 'shellwords'
 #
 # docker_plugin_remove_flags.rb
 #
 module Puppet::Parser::Functions
   # Transforms a hash into a string of docker plugin remove flags
   newfunction(:docker_plugin_enable_flags, type: :rvalue) do |args|
     opts = args[0] || {}
     flags = []
     flags << '--force' if opts['force_remove'] == true
-    flags << "'#{opts['plugin_name']}'" if opts['plugin_name'] && opts['plugin_name'].to_s != 'undef'
+    if opts['plugin_alias'] && opts['plugin_alias'].to_s != 'undef'
+      flags << "'#{opts['plugin_alias']}'"
+    else
+      flags << "'#{opts['plugin_name']}'" if opts['plugin_name'] && opts['plugin_name'].to_s != 'undef'
+    end
     flags.flatten.join(' ')
   end
 end