diff --git a/src/Notification/NotificationActionButton.svelte b/src/Notification/NotificationActionButton.svelte
index 26cf5811..721a4f9e 100644
--- a/src/Notification/NotificationActionButton.svelte
+++ b/src/Notification/NotificationActionButton.svelte
@@ -1,12 +1,12 @@
 <script>
-  import { Button } from "../Button";
+  import Button from "../Button/Button.svelte";
 </script>
 
 <Button
   kind="ghost"
   size="small"
-  class="bx--inline-notification__action-button"
   {...$$restProps}
+  class="bx--inline-notification__action-button {$$restProps.class}"
   on:click
   on:mouseover
   on:mouseenter