diff --git a/src/storm/storage/Scheduler.cpp b/src/storm/storage/Scheduler.cpp
index 6405528a0..a7acff700 100644
--- a/src/storm/storage/Scheduler.cpp
+++ b/src/storm/storage/Scheduler.cpp
@@ -323,9 +323,9 @@ namespace storm {
                 }
             }
             if (numOfSkippedStatesWithUniqueChoice > 0) {
-                stateString << "Skipped " << numOfSkippedStatesWithUniqueChoice << " deterministic states with unique choice." << std::endl;
+                out << "Skipped " << numOfSkippedStatesWithUniqueChoice << " deterministic states with unique choice." << std::endl;
             }
-            stateString << "___________________________________________________________________" << std::endl;
+            out << "___________________________________________________________________" << std::endl;
 
         }
 
diff --git a/src/storm/storage/Scheduler.h b/src/storm/storage/Scheduler.h
index 96819be58..5e8aa3e5e 100644
--- a/src/storm/storage/Scheduler.h
+++ b/src/storm/storage/Scheduler.h
@@ -143,8 +143,9 @@ namespace storm {
              * Prints the scheduler in json format to the given output stream.
              */
             void printJsonToStream(std::ostream& out, std::shared_ptr<storm::models::sparse::Model<ValueType>> model = nullptr, bool skipUniqueChoices = false,  bool skipDontCareStates = false) const;
+            void setPrintUndefinedChoices(bool value);
 
-        private:
+        protected:
             boost::optional<storm::storage::MemoryStructure> memoryStructure;
             std::vector<std::vector<SchedulerChoice<ValueType>>> schedulerChoices;