diff --git a/be2-scala/build.sbt b/be2-scala/build.sbt
index c0fb691dab..906a0a0247 100644
--- a/be2-scala/build.sbt
+++ b/be2-scala/build.sbt
@@ -78,6 +78,7 @@ copyProtocolTask := {
 // Add the copyProtocolTask to compile and test scopes
 (Compile/ compile) := ((Compile/ compile) dependsOn copyProtocolTask).value
 (Test/ test) := ((Test/ test) dependsOn copyProtocolTask).value
+assembly := (assembly dependsOn copyProtocolTask).value
 
 // Setup resource directory for jar assembly
 (Compile/ packageBin/ resourceDirectory) := file(".") / "./src/main/resources"