From 2660b9c5315fcbab8d5c67cac01e59844dc84070 Mon Sep 17 00:00:00 2001
From: Matt Bruce <matt.bruce@verizon.com>
Date: Thu, 14 Sep 2023 12:17:51 -0500
Subject: [PATCH] fixes for button refactor

---
 VDSSample/ViewControllers/ButtonViewController.swift   | 4 ++--
 VDSSample/ViewControllers/TextLinkViewController.swift | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

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(){