diff --git a/build.gradle b/build.gradle
index 20ccc9b609..deca6d4a09 100644
--- a/build.gradle
+++ b/build.gradle
@@ -30,13 +30,13 @@ idea {
 }
 
 group = 'org.neo4j.procedure'
-version = '3.1.0.4'
+version = '3.1.2.5'
 archivesBaseName = 'apoc'
 
 description = """neo4j-apoc-procedures"""
 
 ext {
-    neo4jVersion = "3.1.1"
+    neo4jVersion = "3.1.2"
 }
 
 repositories {
diff --git a/src/main/java/apoc/schema/AssertSchemaProcedure.java b/src/main/java/apoc/schema/AssertSchemaProcedure.java
index 28534d2284..2a5fed1e90 100644
--- a/src/main/java/apoc/schema/AssertSchemaProcedure.java
+++ b/src/main/java/apoc/schema/AssertSchemaProcedure.java
@@ -11,11 +11,12 @@
 import org.neo4j.kernel.api.exceptions.Status;
 import org.neo4j.kernel.api.proc.CallableProcedure;
 import org.neo4j.kernel.api.proc.Context;
-import org.neo4j.kernel.api.proc.Mode;
 import org.neo4j.kernel.api.proc.ProcedureSignature;
 import org.neo4j.kernel.internal.GraphDatabaseAPI;
 import org.neo4j.logging.Log;
 import org.neo4j.procedure.Name;
+import org.neo4j.procedure.Mode;
+
 
 import java.util.*;
 import java.util.concurrent.ExecutionException;