diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 5068a051eb246a24db9bfae4d67a870ea8ea058a..52a7d75c1b1a18f056c3db7a1396c3ead8812c75 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -8,6 +8,7 @@ stages:
   - build_docker
   - test
   - release
+  - demo_runs
 
 build_docker_from_scratch:
   only:
@@ -63,7 +64,7 @@ before_script:
   - export MPLBACKEND="agg"
 
 run_critical_filtering:
-  stage: test
+  stage: demo_runs
   script:
     - ls
     - python setup.py install --user -f
@@ -75,7 +76,7 @@ run_critical_filtering:
       - '*.png'
 
 run_nonlinear_critical_filter:
-  stage: test
+  stage: demo_runs
   script:
     - python setup.py install --user -f
     - python3 setup.py install --user -f
@@ -86,7 +87,7 @@ run_nonlinear_critical_filter:
       - '*.png'
 
 run_nonlinear_wiener_filter:
-  stage: test
+  stage: demo_runs
   script:
     - python setup.py install --user -f
     - python3 setup.py install --user -f
@@ -99,7 +100,7 @@ run_nonlinear_wiener_filter:
       - '*.png'
 
 run_poisson_demo:
-  stage: test
+  stage: demo_runs
   script:
     - python setup.py install --user -f
     - python3 setup.py install --user -f
@@ -110,7 +111,7 @@ run_poisson_demo:
       - '*.png'
 
 run_probing:
-  stage: test
+  stage: demo_runs
   script:
     - python setup.py install --user -f
     - python3 setup.py install --user -f
@@ -121,7 +122,7 @@ run_probing:
       - '*.png'
 
 run_sampling:
-  stage: test
+  stage: demo_runs
   script:
     - python setup.py install --user -f
     - python3 setup.py install --user -f
@@ -132,7 +133,7 @@ run_sampling:
       - '*.png'
 
 run_tomography:
-  stage: test
+  stage: demo_runs
   script:
     - python setup.py install --user -f
     - python3 setup.py install --user -f
@@ -143,7 +144,7 @@ run_tomography:
       - '*.png'
 
 run_wiener_filter_data_space_noiseless:
-  stage: test
+  stage: demo_runs
   script:
     - python setup.py install --user -f
     - python3 setup.py install --user -f
@@ -154,7 +155,7 @@ run_wiener_filter_data_space_noiseless:
       - '*.png'
 
 run_wiener_filter_easy.py:
-  stage: test
+  stage: demo_runs
   script:
     - python setup.py install --user -f
     - python3 setup.py install --user -f
@@ -165,7 +166,7 @@ run_wiener_filter_easy.py:
       - '*.png'
 
 run_wiener_filter_via_curvature.py:
-  stage: test
+  stage: demo_runs
   script:
     - pip install --user numericalunits
     - pip3 install --user numericalunits
@@ -178,7 +179,7 @@ run_wiener_filter_via_curvature.py:
       - '*.png'
 
 run_wiener_filter_via_hamiltonian.py:
-  stage: test
+  stage: demo_runs
   script:
     - python setup.py install --user -f
     - python3 setup.py install --user -f
@@ -189,7 +190,7 @@ run_wiener_filter_via_hamiltonian.py:
       - '*.png'
 
 run_ipynb:
-  stage: test
+  stage: demo_runs
   script:
     - python setup.py install --user -f
     - python3 setup.py install --user -f
diff --git a/nifty5/models/variable.py b/nifty5/models/variable.py
index e6df5603110c798538b3c6276945bc729c322644..d5d361e90859ac39873c37ef6ef35732f9abd19d 100644
--- a/nifty5/models/variable.py
+++ b/nifty5/models/variable.py
@@ -10,7 +10,7 @@ class Variable(Model):
         super(Variable, self).__init__(position)
 
         self._value = position
-        self._gradient = ift.ScalingOperator(1., position.domain)
+        self._gradient = ScalingOperator(1., position.domain)
 
     def at(self, position):
         return self.__class__(position)