diff --git a/src/main/java/seng202/group9/GUI/AirlineGraphController.java b/src/main/java/seng202/group9/GUI/AirlineGraphController.java
index f637b36..1a682bd 100644
--- a/src/main/java/seng202/group9/GUI/AirlineGraphController.java
+++ b/src/main/java/seng202/group9/GUI/AirlineGraphController.java
@@ -19,6 +19,9 @@ public class AirlineGraphController extends Controller{
 
     @Override
     public void load() {
+        if (!checkDataset()){
+            return;
+        }
         dataset = getParent().getCurrentDataset();
         session = getParent().getSession();
     }
diff --git a/src/main/java/seng202/group9/GUI/AirportGraphController.java b/src/main/java/seng202/group9/GUI/AirportGraphController.java
index 7bf6110..6b9d0d1 100644
--- a/src/main/java/seng202/group9/GUI/AirportGraphController.java
+++ b/src/main/java/seng202/group9/GUI/AirportGraphController.java
@@ -17,6 +17,9 @@ public class AirportGraphController extends Controller{
 
     @Override
     public void load() {
+        if (!checkDataset()){
+            return;
+        }
         dataset = getParent().getCurrentDataset();
         session = getParent().getSession();
     }
diff --git a/src/main/java/seng202/group9/GUI/RouteGraphController.java b/src/main/java/seng202/group9/GUI/RouteGraphController.java
index 8815152..f4d74f7 100644
--- a/src/main/java/seng202/group9/GUI/RouteGraphController.java
+++ b/src/main/java/seng202/group9/GUI/RouteGraphController.java
@@ -17,6 +17,9 @@ public class RouteGraphController extends Controller{
 
     @Override
     public void load() {
+        if (!checkDataset()){
+            return;
+        }
         dataset = getParent().getCurrentDataset();
         session = getParent().getSession();
     }