diff --git a/unit_tests/CMakeLists.txt b/unit_tests/CMakeLists.txt
index 852cbd40..18e6ab6d 100644
--- a/unit_tests/CMakeLists.txt
+++ b/unit_tests/CMakeLists.txt
@@ -36,7 +36,7 @@ add_executable(test-main test_main.cpp
 target_link_libraries(test-main KokkosComm gtest gtest_mpi)
 
 add_test(NAME test-main
-         COMMAND mpirun -np 2 ./test-main
+         COMMAND mpirun -np 2 --oversubscribe ./test-main
 )
 
 
diff --git a/unit_tests/test_mpi.cpp b/unit_tests/test_mpi.cpp
index 9363fa1e..2cc5092e 100644
--- a/unit_tests/test_mpi.cpp
+++ b/unit_tests/test_mpi.cpp
@@ -1,5 +1,7 @@
 #include <mpi.h>
 
+#include <iostream>
+
 int main(int argc, char *argv[]) {
   // Initialize MPI before any call to gtest_mpi
   MPI_Init(&argc, &argv);