Update for new sbt-assembly