Add missing includes -- unit tests are apparently useful...