diff --git a/MVMCoreUI/Atomic/Atoms/Buttons/ButtonModel.swift b/MVMCoreUI/Atomic/Atoms/Buttons/ButtonModel.swift
index b15f8adb..94a7778b 100644
--- a/MVMCoreUI/Atomic/Atoms/Buttons/ButtonModel.swift
+++ b/MVMCoreUI/Atomic/Atoms/Buttons/ButtonModel.swift
@@ -9,6 +9,7 @@
 import UIKit
 import VDSColorTokens
 import VDS
+import MVMCore
 
 public typealias FacadeElements = (fill: UIColor?, text: UIColor?, border: UIColor?)
 
diff --git a/MVMCoreUI/Atomic/Atoms/Buttons/PillButton.swift b/MVMCoreUI/Atomic/Atoms/Buttons/PillButton.swift
index c93a22db..645c6602 100644
--- a/MVMCoreUI/Atomic/Atoms/Buttons/PillButton.swift
+++ b/MVMCoreUI/Atomic/Atoms/Buttons/PillButton.swift
@@ -37,6 +37,7 @@ open class PillButton: VDS.Button, MVMCoreUIViewConstrainingProtocol, MFButtonPr
         isEnabled = viewModel.enabled
         size = viewModel.size
         use = viewModel.style ?? .primary
+        surface = viewModel.inverted ? .dark : .light
         if let accessibilityText = viewModel.accessibilityText {
             accessibilityLabel = accessibilityText
         }