diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index a27194d..670e0b7 100644
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -2,8 +2,6 @@
 # CMakeLists.txt - CMake configuration file for epanet/tests
 #
 # Created: February 13, 2018
-# Author:  Constantin Savtchenko
-# Ref: http://neyasystems.com/an-engineers-guide-to-unit-testing-cmake-and-boost-unit-tests/
 #
 # Modified by: Michael E. Tryby
 #              US EPA ORD/NRMRL
diff --git a/tests/test_toolkit.cpp b/tests/test_toolkit.cpp
index 33945c8..f4c760b 100644
--- a/tests/test_toolkit.cpp
+++ b/tests/test_toolkit.cpp
@@ -11,6 +11,7 @@
  ******************************************************************************
 */
 
+#define BOOST_TEST_MAIN
 #define BOOST_TEST_MODULE toolkit
 
 #include <boost/test/unit_test.hpp>