diff --git a/python/CMakeLists.txt b/python/CMakeLists.txt
index 4261f5cb4..1134c1c8d 100644
--- a/python/CMakeLists.txt
+++ b/python/CMakeLists.txt
@@ -124,6 +124,7 @@ if (BUILD_TESTING AND NOT WIN32)
   )
 
   set(python_tests
+    pyAirFlow_TEST
     pyAirPressure_TEST
     pyAirSpeed_TEST
     pyAltimeter_TEST
diff --git a/python/src/sdf/pyAirFlow.cc b/python/src/sdf/pyAirFlow.cc
index 88144c54a..9519a74a7 100644
--- a/python/src/sdf/pyAirFlow.cc
+++ b/python/src/sdf/pyAirFlow.cc
@@ -43,7 +43,7 @@ void defineAirFlow(pybind11::object module)
     .def("set_speed_noise",
          &sdf::AirFlow::SetSpeedNoise,
          "Set the noise values related to the speed data.")
-    .def("dir_noise", &sdf::AirFlow::DirectionNoise,
+    .def("direction_noise", &sdf::AirFlow::DirectionNoise,
          "Get the direction noise values.")
     .def("set_direction_noise",
          &sdf::AirFlow::SetDirectionNoise,
diff --git a/python/test/pyAirFlow_TEST.py b/python/test/pyAirFlow_TEST.py
index 68c2d0c92..12c70f2f9 100644
--- a/python/test/pyAirFlow_TEST.py
+++ b/python/test/pyAirFlow_TEST.py
@@ -48,6 +48,7 @@ def test_set(self):
         speed_noise.set_precision(8.9)
 
         air.set_direction_noise(dir_noise)
+        air.set_speed_noise(speed_noise)
         self.assertEqual(dir_noise, air.direction_noise())
         self.assertEqual(speed_noise, air.speed_noise())