diff --git a/src/testUtils.ts b/src/testUtils.ts
index 42b5ef4..1fbfc3b 100644
--- a/src/testUtils.ts
+++ b/src/testUtils.ts
@@ -9,6 +9,7 @@
 
 import { applyCodeCoverageToAllEditors } from './goCover';
 import { toolExecutionEnvironment } from './goEnv';
+import { getCurrentPackage } from './goModules';
 import { GoDocumentSymbolProvider } from './goOutline';
 import { getNonVendorPackages } from './goPackages';
 import {
@@ -303,6 +304,7 @@
 		let pkgMapPromise: Promise<Map<string, string> | null> = Promise.resolve(null);
 
 		if (testconfig.isMod) {
+			getCurrentPackagePromise = getCurrentPackage(testconfig.dir);
 			// We need the mapping to get absolute paths for the files in the test output.
 			pkgMapPromise = getNonVendorPackages(testconfig.dir, !!testconfig.includeSubDirectories);
 		} else {  // GOPATH mode