diff --git a/kotlin/builder/src/io/bazel/kotlin/builder/KotlinBuilder.kt b/kotlin/builder/src/io/bazel/kotlin/builder/KotlinBuilder.kt
index 0f20535..b56b8dd 100644
--- a/kotlin/builder/src/io/bazel/kotlin/builder/KotlinBuilder.kt
+++ b/kotlin/builder/src/io/bazel/kotlin/builder/KotlinBuilder.kt
@@ -67,7 +67,7 @@
                 it.jarFiles.addAll(command.inputs.sourceJarsList.map { Paths.get(it) })
                 it.execute()
             }.let {
-                command.expandWithSources(command, it.sourcesList.iterator())
+                command.expandWithSources(it.sourcesList.iterator())
             }
         }
 
diff --git a/kotlin/builder/src/io/bazel/kotlin/builder/tasks/jvm/KotlinJvmTaskExecutor.kt b/kotlin/builder/src/io/bazel/kotlin/builder/tasks/jvm/KotlinJvmTaskExecutor.kt
index fbf7655..505e77b 100644
--- a/kotlin/builder/src/io/bazel/kotlin/builder/tasks/jvm/KotlinJvmTaskExecutor.kt
+++ b/kotlin/builder/src/io/bazel/kotlin/builder/tasks/jvm/KotlinJvmTaskExecutor.kt
@@ -58,7 +58,7 @@
                     .filter { it.isFile }
                     .map { it.path }
                     .iterator()
-                    .let { command.expandWithGeneratedSources(command, it) }
+                    .let { command.expandWithGeneratedSources(it) }
             } else {
                 command
             }
diff --git a/kotlin/builder/src/io/bazel/kotlin/builder/utils/TaskUtils.kt b/kotlin/builder/src/io/bazel/kotlin/builder/utils/TaskUtils.kt
index 4ec19b4..2eb749d 100644
--- a/kotlin/builder/src/io/bazel/kotlin/builder/utils/TaskUtils.kt
+++ b/kotlin/builder/src/io/bazel/kotlin/builder/utils/TaskUtils.kt
@@ -19,10 +19,9 @@
 
 
 fun KotlinModel.CompilationTask.expandWithSources(
-    command: KotlinModel.CompilationTask,
     sources: Iterator<String>
 ): KotlinModel.CompilationTask =
-    command.updateBuilder { builder ->
+    updateBuilder { builder ->
         sources.partitionSources(
             { builder.inputsBuilder.addKotlinSources(it) },
             { builder.inputsBuilder.addJavaSources(it) })
@@ -30,10 +29,9 @@
 
 
 fun KotlinModel.CompilationTask.expandWithGeneratedSources(
-    command: KotlinModel.CompilationTask,
     sources: Iterator<String>
 ): KotlinModel.CompilationTask =
-    command.updateBuilder { builder ->
+    updateBuilder { builder ->
         sources.partitionSources(
             { builder.inputsBuilder.addGeneratedKotlinSources(it) },
             { builder.inputsBuilder.addGeneratedJavaSources(it) })