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