diff --git a/alfa-client/libs/design-component/src/lib/open-dialog-button/open-dialog-button.component.ts b/alfa-client/libs/design-component/src/lib/open-dialog-button/open-dialog-button.component.ts
index 833c45112be776c724cd1bfa8314f89b5f873335..ae0e20c3b018b6a00332bbf4d4aa7680fe300546 100644
--- a/alfa-client/libs/design-component/src/lib/open-dialog-button/open-dialog-button.component.ts
+++ b/alfa-client/libs/design-component/src/lib/open-dialog-button/open-dialog-button.component.ts
@@ -16,6 +16,7 @@ import { first } from 'rxjs';
     [text]="label"
     [size]="size"
     [dataTestId]="dataTestId"
+    [dataTestClass]="dataTestClass"
     data-test-id="open-dialog"
   >
     <ng-container icon>
@@ -34,6 +35,7 @@ export class OpenDialogButtonComponent {
 
   @Input() label: string;
   @Input() dataTestId: string;
+  @Input() dataTestClass: string;
   @Input() variant: ButtonVariants['variant'] = 'primary';
   @Input() dialogData: any;
   @Input() size: ButtonVariants['size'];
diff --git a/alfa-client/libs/design-system/src/lib/button/button.component.ts b/alfa-client/libs/design-system/src/lib/button/button.component.ts
index c0f7892c3c2c5184571b175dbff2326591415495..47e76172adb5e6f9600be54291ee79a841defb78 100644
--- a/alfa-client/libs/design-system/src/lib/button/button.component.ts
+++ b/alfa-client/libs/design-system/src/lib/button/button.component.ts
@@ -96,6 +96,7 @@ export type ButtonVariants = VariantProps<typeof buttonVariants>;
     [attr.aria-disabled]="isDisabled"
     [attr.aria-label]="text"
     [attr.data-test-id]="dataTestId"
+    [attr.data-test-class]="dataTestClass"
     (click)="clickEmitter.emit()"
   >
     <ng-content *ngIf="!isLoading" select="[icon]"></ng-content>
@@ -106,6 +107,7 @@ export type ButtonVariants = VariantProps<typeof buttonVariants>;
 export class ButtonComponent {
   @Input() text: string = '';
   @Input() dataTestId: string = '';
+  @Input() dataTestClass: string = '';
   @Input() disabled: boolean = false;
   @Input() isLoading: boolean = false;
   @Input({ transform: booleanAttribute }) destructive: boolean = false;