Fix missing import at assembly.sbt