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