diff --git a/Chapter_6/Ch6_final.py b/Chapter_6/Ch6_final.py
index 3c06055d5..d23f9bc91 100644
--- a/Chapter_6/Ch6_final.py
+++ b/Chapter_6/Ch6_final.py
@@ -1,6 +1,6 @@
 import numpy as np
 import nnfs
-from nnfs.datasets import spiral_data
+from nnfs.datasets import vertical_data
 
 nnfs.init()