diff --git a/src/Ryujinx.Graphics.Shader/CodeGen/Msl/CodeGenContext.cs b/src/Ryujinx.Graphics.Shader/CodeGen/Msl/CodeGenContext.cs
index 79c13964ca..0ae6313eb7 100644
--- a/src/Ryujinx.Graphics.Shader/CodeGen/Msl/CodeGenContext.cs
+++ b/src/Ryujinx.Graphics.Shader/CodeGen/Msl/CodeGenContext.cs
@@ -9,7 +9,7 @@ namespace Ryujinx.Graphics.Shader.CodeGen.Msl
         public const string Tab = "    ";
 
         // The number of additional arguments that every function (except for the main one) must have (for instance support_buffer)
-        public const int AdditionalArgCount = 2;
+        public const int AdditionalArgCount = 1;
 
         public StructuredFunction CurrentFunction { get; set; }
 
diff --git a/src/Ryujinx.Graphics.Shader/CodeGen/Msl/Instructions/InstGenCall.cs b/src/Ryujinx.Graphics.Shader/CodeGen/Msl/Instructions/InstGenCall.cs
index 5df3aa2820..c063ff4589 100644
--- a/src/Ryujinx.Graphics.Shader/CodeGen/Msl/Instructions/InstGenCall.cs
+++ b/src/Ryujinx.Graphics.Shader/CodeGen/Msl/Instructions/InstGenCall.cs
@@ -13,13 +13,22 @@ namespace Ryujinx.Graphics.Shader.CodeGen.Msl.Instructions
             var functon = context.GetFunction(funcId.Value);
 
             int argCount = operation.SourcesCount - 1;
-            string[] args = new string[argCount + CodeGenContext.AdditionalArgCount];
+            int additionalArgCount = CodeGenContext.AdditionalArgCount + (context.Definitions.Stage != ShaderStage.Compute ? 1 : 0);
+
+            string[] args = new string[argCount + additionalArgCount];
 
             // Additional arguments
-            args[0] = "in";
-            args[1] = "support_buffer";
+            if (context.Definitions.Stage != ShaderStage.Compute)
+            {
+                args[0] = "in";
+                args[1] = "support_buffer";
+            }
+            else
+            {
+                args[0] = "support_buffer";
+            }
 
-            int argIndex = CodeGenContext.AdditionalArgCount;
+            int argIndex = additionalArgCount;
             for (int i = 0; i < argCount; i++)
             {
                 args[argIndex++] = GetSourceExpr(context, operation.GetSource(i + 1), functon.GetArgumentType(i));
diff --git a/src/Ryujinx.Graphics.Shader/CodeGen/Msl/MslGenerator.cs b/src/Ryujinx.Graphics.Shader/CodeGen/Msl/MslGenerator.cs
index 87512a9617..d68ebb9ebd 100644
--- a/src/Ryujinx.Graphics.Shader/CodeGen/Msl/MslGenerator.cs
+++ b/src/Ryujinx.Graphics.Shader/CodeGen/Msl/MslGenerator.cs
@@ -63,15 +63,22 @@ namespace Ryujinx.Graphics.Shader.CodeGen.Msl
             ShaderStage stage,
             bool isMainFunc = false)
         {
-            int additionalArgCount = isMainFunc ? 0 : CodeGenContext.AdditionalArgCount;
+            int additionalArgCount = isMainFunc ? 0 : CodeGenContext.AdditionalArgCount + (context.Definitions.Stage != ShaderStage.Compute ? 1 : 0);
 
             string[] args = new string[additionalArgCount + function.InArguments.Length + function.OutArguments.Length];
 
             // All non-main functions need to be able to access the support_buffer as well
             if (!isMainFunc)
             {
-                args[0] = "FragmentIn in";
-                args[1] = "constant Struct_support_buffer* support_buffer";
+                if (stage != ShaderStage.Compute)
+                {
+                    args[0] = stage == ShaderStage.Vertex ? "VertexIn in" : "FragmentIn in";
+                    args[1] = "constant Struct_support_buffer* support_buffer";
+                }
+                else
+                {
+                    args[0] = "constant Struct_support_buffer* support_buffer";
+                }
             }
 
             int argIndex = additionalArgCount;