added missing includes for tests