diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 8e5f5f52a4a73e8feb9cefa3b491df9e5aaf2713..e067b025d04ce2791489a56fbb079b49c7672c70 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -24,6 +24,7 @@ build and upload the contents:
 
   script:
     # Compile lectures
+    - python code/band_structures.py
     - gitbook install
     - gitbook build
     - "rsync -rv _book solidstate@tnw-tn1.tudelft.net:test"
diff --git a/code/band_structures.py b/code/band_structures.py
index 58482f611eed24470439284f85d6abbb037d7044..c09ccd15b35e978dfdef03ff2c8d7e11110f1b2f 100644
--- a/code/band_structures.py
+++ b/code/band_structures.py
@@ -120,5 +120,5 @@ def phonons_1d_2masses():
 
 
 def main():
-    os.chdir('../figures')
+    os.chdir('figures')
     fermi_dirac()
diff --git a/figures/einstein.png b/figures/einstein.png
new file mode 100644
index 0000000000000000000000000000000000000000..14efea0b2e004d14b84626208798236e22e5c361
Binary files /dev/null and b/figures/einstein.png differ