diff --git a/VDSSample/ViewControllers/BadgeViewController.swift b/VDSSample/ViewControllers/BadgeViewController.swift
index 1ff1d0d..7b101df 100644
--- a/VDSSample/ViewControllers/BadgeViewController.swift
+++ b/VDSSample/ViewControllers/BadgeViewController.swift
@@ -40,7 +40,7 @@ class  BadgeViewController: BaseViewController {
     lazy var fillColorPickerSelectorView = {
         PickerSelectorView(title: "red",
                             picker: self.picker,
-                            items: BadgeFillColor.allCases)
+                           items: Badge.FillColor.allCases)
     }()
     
     lazy var numberOfLinesPickerSelectorView = {