diff --git a/packages/superset-ui-core/test/query/getColumnLabel.test.ts b/packages/superset-ui-core/test/query/getColumnLabel.test.ts
index aadde6e4b5..865463481c 100644
--- a/packages/superset-ui-core/test/query/getColumnLabel.test.ts
+++ b/packages/superset-ui-core/test/query/getColumnLabel.test.ts
@@ -16,7 +16,7 @@
  * specific language governing permissions and limitations
  * under the License.
  */
-import { getColumnLabel } from '@superset-ui/core';
+import { getColumnLabel } from '@superset-ui/core/src/query';
 
 describe('getColumnLabel', () => {
   it('should handle physical column', () => {
diff --git a/packages/superset-ui-core/test/query/getMetricLabel.test.ts b/packages/superset-ui-core/test/query/getMetricLabel.test.ts
index 057c3225ea..434eb72155 100644
--- a/packages/superset-ui-core/test/query/getMetricLabel.test.ts
+++ b/packages/superset-ui-core/test/query/getMetricLabel.test.ts
@@ -16,7 +16,7 @@
  * specific language governing permissions and limitations
  * under the License.
  */
-import { getMetricLabel } from '@superset-ui/core';
+import { getMetricLabel } from '@superset-ui/core/src/query';
 
 describe('getMetricLabel', () => {
   it('should handle predefined metric name', () => {