diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index ede382ca941c505df47e4cf18089a6fa2ecf7944..e51b1f329eb701f7efb85aed0e92eb3181884be1 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -28,7 +28,7 @@ typechecker:
 
 build-documentation:
   stage: test
-  image: i10git.cs.fau.de:5005/pycodegen/pycodegen/documentation
+  image: i10git.cs.fau.de:5005/pycodegen/pycodegen/full
   script:
     - pip install mkdocs mkdocs-material mkdocstrings[python]
     - mkdocs build -v
diff --git a/src/pystencilssfg/__init__.py b/src/pystencilssfg/__init__.py
index ae3cd1a774207f5a8d2515fce5f98254411ce366..96c285db2461ef2dea5bbc6880c4e6c3f5eedcdd 100644
--- a/src/pystencilssfg/__init__.py
+++ b/src/pystencilssfg/__init__.py
@@ -2,7 +2,7 @@ from .generator import SourceFileGenerator
 from .composer import SfgComposer
 
 __all__ = [
-    "SourceFileGenerator", "SfgComposer",
+    "SourceFileGenerator", "SfgComposer"
 ]
 
 from . import _version