diff --git a/src/test/scala/service/ServiceSpecBase.scala b/src/test/scala/service/ServiceSpecBase.scala
index be69750..a87bb58 100644
--- a/src/test/scala/service/ServiceSpecBase.scala
+++ b/src/test/scala/service/ServiceSpecBase.scala
@@ -16,7 +16,7 @@
       val (url, user, pass) = (DatabaseConfig.url(Some(dir.toString)), DatabaseConfig.user, DatabaseConfig.password)
       org.h2.Driver.load()
       using(DriverManager.getConnection(url, user, pass)){ conn =>
-        servlet.AutoUpdate.versions.reverse.foreach(_.update(conn))
+        servlet.AutoUpdate.versions.reverse.foreach(_.update(conn, Thread.currentThread.getContextClassLoader))
       }
       Database.forURL(url, user, pass).withSession { session =>
         action(session)
diff --git a/src/test/scala/ssh/GitCommandSpec.scala b/src/test/scala/ssh/GitCommandSpec.scala
index c5c4630..6c571ae 100644
--- a/src/test/scala/ssh/GitCommandSpec.scala
+++ b/src/test/scala/ssh/GitCommandSpec.scala
@@ -7,7 +7,7 @@
 
 class GitCommandFactorySpec extends Specification with Mockito {
 
-  val factory = new GitCommandFactory(mock[ServletContext], "http://localhost:8080")
+  val factory = new GitCommandFactory("http://localhost:8080")
 
   "createCommand" should {
     "returns GitReceivePack when command is git-receive-pack" in {