diff --git a/src/storm-dft/storage/dft/DFTBuilder.cpp b/src/storm-dft/storage/dft/DFTBuilder.cpp
index ed35b176d..35ed767b8 100644
--- a/src/storm-dft/storage/dft/DFTBuilder.cpp
+++ b/src/storm-dft/storage/dft/DFTBuilder.cpp
@@ -91,7 +91,9 @@ namespace storm {
 
             // Set layout info
             for (auto& elem : mElements) {
-                dft.setElementLayoutInfo(elem.second->id(), mLayoutInfo.at(elem.first));
+                if(mLayoutInfo.count(elem.first) > 0) {
+                    dft.setElementLayoutInfo(elem.second->id(), mLayoutInfo.at(elem.first));
+                }
             }
 
             return dft;