diff --git a/Example/Sources/PetSelectorBulletinPage.swift b/Example/Sources/PetSelectorBulletinPage.swift
index a1ccbbe..0d47aac 100644
--- a/Example/Sources/PetSelectorBulletinPage.swift
+++ b/Example/Sources/PetSelectorBulletinPage.swift
@@ -153,7 +153,7 @@ class PetSelectorBulletinPage: BulletinItem {
             button.accessibilityTraits &= ~UIAccessibilityTraitSelected
         }
 
-        let buttonContainer = ContainerView<UIButton>(button)
+        let buttonContainer = ContainerView(button)
         buttonContainer.layer.cornerRadius = 12
         buttonContainer.layer.borderWidth = 2
 
diff --git a/Sources/BulletinInterfaceFactory.swift b/Sources/BulletinInterfaceFactory.swift
index 91b543a..fc73746 100644
--- a/Sources/BulletinInterfaceFactory.swift
+++ b/Sources/BulletinInterfaceFactory.swift
@@ -117,7 +117,7 @@ public class BulletinInterfaceFactory {
         actionButton.layer.cornerRadius = 12
         actionButton.clipsToBounds = true
 
-        let actionContainer = ContainerView<HighlightButton>(actionButton)
+        let actionContainer = ContainerView(actionButton)
         actionContainer.heightAnchor.constraint(equalToConstant: 55).isActive = true
 
         return actionContainer