diff --git a/src/core/transformation.cpp b/src/core/transformation.cpp
index 984c0e8..6db4330 100644
--- a/src/core/transformation.cpp
+++ b/src/core/transformation.cpp
@@ -1,4 +1,5 @@
 #include "transformation.h"
+#include "storm/models/symbolic/StandardRewardModel.h"
 
 void define_transformation(py::module& m) {
     // Transform model