diff --git a/tasks/main.yml b/tasks/main.yml
index c3a554b..50c6b13 100644
--- a/tasks/main.yml
+++ b/tasks/main.yml
@@ -32,8 +32,8 @@
 
 - lineinfile:
     path: "/root/.bashrc"
-    regexp: '^alias su='
-    line: "alias su=\"machinectl shell --uid\""
+    regexp: '^su() '
+    line: "su() { if [[ $1 == \"-\" ]]; then command machinectl shell --uid \"$2\"; else command machinectl shell --uid \"$1\"; fi; }"
  
 - name: Enable permissive_container_t SELinux Context
   selinux_permissive: