diff --git a/packages/artifact/src/internal/shared/util.ts b/packages/artifact/src/internal/shared/util.ts
index 50844a40..bc98abde 100644
--- a/packages/artifact/src/internal/shared/util.ts
+++ b/packages/artifact/src/internal/shared/util.ts
@@ -1,6 +1,5 @@
 import {getRuntimeToken} from './config'
 import jwt_decode from 'jwt-decode'
-import {Timestamp} from '../../generated'
 
 export interface BackendIds {
   workflowRunBackendId: string
@@ -64,14 +63,3 @@ export function getBackendIdsFromToken(): BackendIds {
 
   throw InvalidJwtError
 }
-
-export function getExpiration(retentionDays?: number): Timestamp | undefined {
-  if (!retentionDays) {
-    return undefined
-  }
-
-  const expirationDate = new Date()
-  expirationDate.setDate(expirationDate.getDate() + retentionDays)
-
-  return Timestamp.fromDate(expirationDate)
-}
diff --git a/packages/artifact/src/internal/upload/upload-artifact.ts b/packages/artifact/src/internal/upload/upload-artifact.ts
index 8ad52390..087493a3 100644
--- a/packages/artifact/src/internal/upload/upload-artifact.ts
+++ b/packages/artifact/src/internal/upload/upload-artifact.ts
@@ -8,8 +8,8 @@ import {
   getUploadZipSpecification,
   validateRootDirectory
 } from './upload-zip-specification'
-import {getBackendIdsFromToken, getExpiration} from '../shared/util'
-import {CreateArtifactRequest} from 'src/generated'
+import {getBackendIdsFromToken} from '../shared/util'
+import {CreateArtifactRequest, Timestamp} from 'src/generated'
 
 export async function uploadArtifact(
   name: string,
@@ -91,3 +91,14 @@ export async function uploadArtifact(
 
   return uploadResponse
 }
+
+function getExpiration(retentionDays?: number): Timestamp | undefined {
+  if (!retentionDays) {
+    return undefined
+  }
+
+  const expirationDate = new Date()
+  expirationDate.setDate(expirationDate.getDate() + retentionDays)
+
+  return Timestamp.fromDate(expirationDate)
+}