diff --git a/GradientLoadingBar/Classes/Misc/Constants.swift b/GradientLoadingBar/Classes/Misc/Constants.swift
index d9d34b6..e7dcf2b 100644
--- a/GradientLoadingBar/Classes/Misc/Constants.swift
+++ b/GradientLoadingBar/Classes/Misc/Constants.swift
@@ -6,7 +6,7 @@
 //  Copyright © 2019 Felix Mau. All rights reserved.
 //
 
-import Foundation
+import UIKit
 
 extension UIColor {
     /// Default colors for components.
diff --git a/GradientLoadingBar/Classes/ViewModel/GradientActivityIndicatorViewModel.swift b/GradientLoadingBar/Classes/ViewModel/GradientActivityIndicatorViewModel.swift
index 8ec2e84..c14a0f0 100644
--- a/GradientLoadingBar/Classes/ViewModel/GradientActivityIndicatorViewModel.swift
+++ b/GradientLoadingBar/Classes/ViewModel/GradientActivityIndicatorViewModel.swift
@@ -6,7 +6,7 @@
 //  Copyright © 2019 Felix Mau. All rights reserved.
 //
 
-import Foundation
+import UIKit
 import LightweightObservable
 
 // MARK: - Types
diff --git a/GradientLoadingBar/Classes/ViewModel/GradientLoadingBarViewModel.swift b/GradientLoadingBar/Classes/ViewModel/GradientLoadingBarViewModel.swift
index c1e5454..d720e90 100644
--- a/GradientLoadingBar/Classes/ViewModel/GradientLoadingBarViewModel.swift
+++ b/GradientLoadingBar/Classes/ViewModel/GradientLoadingBarViewModel.swift
@@ -6,7 +6,7 @@
 //  Copyright © 2017 Felix Mau. All rights reserved.
 //
 
-import Foundation
+import UIKit
 import LightweightObservable
 
 /// This view model checks for the availability of the key-window,