diff --git a/documentation/documentation.gradle b/documentation/documentation.gradle
index d67d2568aa..f146884876 100644
--- a/documentation/documentation.gradle
+++ b/documentation/documentation.gradle
@@ -55,6 +55,7 @@ dependencies {
 	testCompile( libraries.mockito )
 	testCompile( libraries.mockito_inline )
 
+	testRuntime(libraries.wildfly_transaction_client)
 	testRuntime( libraries.h2 )
 	testRuntime( libraries.hsqldb )
 	testRuntime( libraries.postgresql )
diff --git a/hibernate-core/hibernate-core.gradle b/hibernate-core/hibernate-core.gradle
index 349b0b1fa9..e0712de971 100644
--- a/hibernate-core/hibernate-core.gradle
+++ b/hibernate-core/hibernate-core.gradle
@@ -52,7 +52,6 @@ dependencies {
     //      3) hibernate-commons-annotations
     compile( libraries.dom4j )
     compile( libraries.commons_annotations )
-    compile(libraries.wildfly_transaction_client)
     antlr( libraries.antlr )
 
     provided( libraries.jacc )
@@ -90,8 +89,9 @@ dependencies {
     testRuntime( libraries.javassist )
     testRuntime( libraries.byteBuddy )
     testRuntime( libraries.weld )
+    testRuntime(libraries.wildfly_transaction_client)
 
-    testCompile( project( ':hibernate-jpamodelgen' ) )
+	testCompile( project( ':hibernate-jpamodelgen' ) )
 
     testCompile libraries.arquillian_junit_container
     testCompile libraries.arquillian_protocol_servlet