diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/Session.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/Session.java
index 5b9f2f1985e..8e65425cf35 100644
--- a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/Session.java
+++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/Session.java
@@ -703,7 +703,7 @@ public synchronized void restore(String prefix) {
     markAllInitializersAsRan();
   }
 
-  Graph graph() {
+  public Graph graph() {
     return graph;
   }