diff --git a/packages/objectloader2/src/index.ts b/packages/objectloader2/src/index.ts
index b8dba46b8..c5e6b6e42 100644
--- a/packages/objectloader2/src/index.ts
+++ b/packages/objectloader2/src/index.ts
@@ -1,3 +1,4 @@
 export { ObjectLoader2 } from './core/objectLoader2.js'
 export { ObjectLoader2Factory } from './core/objectLoader2Factory.js'
 export { getFeatureFlag, ObjectLoader2Flags } from './types/functions.js'
+export { ObjectAttributeMask } from './types/types.js'
diff --git a/packages/viewer/src/modules/loaders/Speckle/SpeckleLoader.ts b/packages/viewer/src/modules/loaders/Speckle/SpeckleLoader.ts
index 03a199986..6940b275a 100644
--- a/packages/viewer/src/modules/loaders/Speckle/SpeckleLoader.ts
+++ b/packages/viewer/src/modules/loaders/Speckle/SpeckleLoader.ts
@@ -7,7 +7,8 @@ import {
   getFeatureFlag,
   ObjectLoader2Flags,
   ObjectLoader2,
-  ObjectLoader2Factory
+  ObjectLoader2Factory,
+  ObjectAttributeMask
 } from '@speckle/objectloader2'
 import { TIME_MS } from '@speckle/shared'
 
@@ -33,7 +34,8 @@ export class SpeckleLoader extends Loader {
     authToken?: string,
     enableCaching?: boolean,
     resourceData?: unknown,
-    logger?: (message?: string, ...args: unknown[]) => void
+    logger?: (message?: string, ...args: unknown[]) => void,
+    attributeMask?: ObjectAttributeMask
   ) {
     super(resource, resourceData)
     this.tree = targetTree
@@ -43,7 +45,8 @@ export class SpeckleLoader extends Loader {
         resource,
         authToken,
         enableCaching,
-        resourceData
+        resourceData,
+        attributeMask
       )
     } catch (e) {
       Logger.error(e)
@@ -57,7 +60,8 @@ export class SpeckleLoader extends Loader {
     resource: string,
     authToken?: string,
     _enableCaching?: boolean,
-    resourceData?: unknown
+    resourceData?: unknown,
+    attributeMask?: ObjectAttributeMask
   ): ObjectLoader2 {
     resourceData
     let token = undefined
@@ -92,7 +96,8 @@ export class SpeckleLoader extends Loader {
       serverUrl,
       streamId,
       objectId,
-      token
+      token,
+      attributeMask
     })
   }