diff --git a/src/mock.cpp b/src/mock.cpp
index 1caa36a..33d208d 100644
--- a/src/mock.cpp
+++ b/src/mock.cpp
@@ -4,13 +4,10 @@
 #ifdef STA_SPATZ_ENABLED
 
 #include <sta/debug/spatz.hpp>
+#include <sta/debug/assert.hpp>
+#include <sta/lang.hpp>
 
-/**
- * @brief SPATZ IDs for the pressure and temperature data.
- * 
- */
-#define STA_MS56XX_PRESSURE_SPATZ_ID        1
-#define STA_MS56XX_TEMPERATURE_SPATZ_ID     2
+#include <math.h>
 
 
 namespace sta
@@ -157,4 +154,4 @@ namespace sta
     }
 } // namespace sta
 
-#endif // STA_SPATZ_ENABLED
\ No newline at end of file
+#endif // STA_SPATZ_ENABLED