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 09dc45136a8..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,11 +703,11 @@ public synchronized void restore(String prefix) {
     markAllInitializersAsRan();
   }
 
-  Graph graph() {
+  public Graph graph() {
     return graph;
   }
 
-  public final Graph graph;
+  private final Graph graph;
   private final Graph.Reference graphRef;
 
   private final Object nativeHandleLock = new Object();