diff --git a/bfps/DNS.py b/bfps/DNS.py
index b17d2611a13b363dc4740b0ae0d2f0e4fc6084b6..f24cd59b8d2edd3c6460996cac9c9bb7f40eca00 100644
--- a/bfps/DNS.py
+++ b/bfps/DNS.py
@@ -74,6 +74,7 @@ class DNS(_code):
                 '"scope_timer.hpp"',
                 '"fftw_interface.hpp"',
                 '"full_code/main_code.hpp"',
+                '<cmath>',
                 '<iostream>',
                 '<hdf5.h>',
                 '<string>',
diff --git a/tests/test_DNS.py b/tests/test_DNS.py
index 3ff6596330e2192bb919dd9611baff03f56fd473..bac769522743232120dea26a55e0fa2fedbb480f 100644
--- a/tests/test_DNS.py
+++ b/tests/test_DNS.py
@@ -2,7 +2,7 @@ from bfps.DNS import DNS
 
 
 def main():
-    c = DNS()
+    c = DNS(dns_type = 'NSVEp')
     c.write_src()
     c.compile_code()
     return None