Fixed minor bug that prevents unit tests from running.