diff --git a/Softeq.XToolkit.WhiteLabel.iOS/ViewControllerBase.cs b/Softeq.XToolkit.WhiteLabel.iOS/ViewControllerBase.cs
index 08673a14..5b43e571 100644
--- a/Softeq.XToolkit.WhiteLabel.iOS/ViewControllerBase.cs
+++ b/Softeq.XToolkit.WhiteLabel.iOS/ViewControllerBase.cs
@@ -7,6 +7,7 @@
 using Softeq.XToolkit.Bindings.Abstract;
 using Softeq.XToolkit.Bindings.Extensions;
 using Softeq.XToolkit.WhiteLabel.Mvvm;
+using Softeq.XToolkit.WhiteLabel.Navigation;
 using UIKit;
 
 namespace Softeq.XToolkit.WhiteLabel.iOS
@@ -120,7 +121,7 @@ private void InitViewController()
 
         private void CloseDialogIfNeeded()
         {
-            if (ViewModel is DialogViewModelBase dialogViewModel)
+            if (ViewModel is IDialogViewModel dialogViewModel)
             {
                 if (IsBeingDismissed || IsMovingFromParentViewController)
                 {