diff --git a/src/storm/storage/jani/Assignment.cpp b/src/storm/storage/jani/Assignment.cpp
index 203d92933..19655a132 100644
--- a/src/storm/storage/jani/Assignment.cpp
+++ b/src/storm/storage/jani/Assignment.cpp
@@ -43,7 +43,11 @@ namespace storm  {
         int64_t Assignment::getLevel() const {
             return level;
         }
-        
+
+        void Assignment::setLevel(int64_t level) const {
+            this->level = level;
+        }
+
         bool Assignment::isLinear() const {
             storm::expressions::LinearityCheckVisitor linearityChecker;
             return linearityChecker.check(this->getAssignedExpression(), true);
diff --git a/src/storm/storage/jani/Assignment.h b/src/storm/storage/jani/Assignment.h
index 46c0a467e..a4e2dbd23 100644
--- a/src/storm/storage/jani/Assignment.h
+++ b/src/storm/storage/jani/Assignment.h
@@ -51,7 +51,12 @@ namespace storm {
              * Retrieves the level of the assignment.
              */
             int64_t getLevel() const;
-            
+
+            /*!
+             * Sets the level
+             */
+            void setLevel(int64_t level);
+
             /*!
              * Checks the assignment for linearity.
              */