diff --git a/VDSSample/ViewControllers/ButtonViewController.swift b/VDSSample/ViewControllers/ButtonViewController.swift
index d43f9ec..1b3973f 100644
--- a/VDSSample/ViewControllers/ButtonViewController.swift
+++ b/VDSSample/ViewControllers/ButtonViewController.swift
@@ -21,7 +21,7 @@ class ButtonViewController: BaseViewController<Button> {
     lazy var buttonSizePickerSelectorView = {
         PickerSelectorView(title: "",
                             picker: self.picker,
-                            items: ButtonSize.allCases)
+                           items: Button.Size.allCases)
     }()
 
     var label = Label()
@@ -77,7 +77,7 @@ class ButtonViewController: BaseViewController<Button> {
         textField.text = component.text
         usePickerSelectorView.text = component.use.rawValue
         widthTextField.text = ""
-        buttonSizePickerSelectorView.text = ButtonSize.large.rawValue
+        buttonSizePickerSelectorView.text = component.size.rawValue
     }
       
     func setupPicker(){
diff --git a/VDSSample/ViewControllers/TextLinkViewController.swift b/VDSSample/ViewControllers/TextLinkViewController.swift
index c20f09f..f5717e9 100644
--- a/VDSSample/ViewControllers/TextLinkViewController.swift
+++ b/VDSSample/ViewControllers/TextLinkViewController.swift
@@ -15,7 +15,7 @@ class TextLinkViewController: BaseViewController<TextLink> {
     lazy var buttonSizePickerSelectorView = {
         PickerSelectorView(title: "",
                             picker: self.picker,
-                            items: ButtonSize.allCases)
+                           items: TextLink.Size.allCases)
     }()
 
     var label = Label()
@@ -59,7 +59,7 @@ class TextLinkViewController: BaseViewController<TextLink> {
         surfacePickerSelectorView.text = component.surface.rawValue
         disabledSwitch.isOn = !component.isEnabled
         textField.text = component.text
-        buttonSizePickerSelectorView.text = ButtonSize.large.rawValue
+        buttonSizePickerSelectorView.text = component.size.rawValue
     }
       
     func setupPicker(){