diff --git a/src/main/java/com/alivanov/intellij/plugins/liquigen/LiquibaseWrapper.java b/src/main/java/com/alivanov/intellij/plugins/liquigen/LiquibaseWrapper.java
index 7032b20..74f9a4c 100644
--- a/src/main/java/com/alivanov/intellij/plugins/liquigen/LiquibaseWrapper.java
+++ b/src/main/java/com/alivanov/intellij/plugins/liquigen/LiquibaseWrapper.java
@@ -3,11 +3,11 @@
 import com.alivanov.intellij.plugins.liquigen.command.AuthorParametrizedDiffToChangeLogCommand;
 import com.intellij.database.dataSource.DatabaseConnection;
 import com.intellij.database.dataSource.DatabaseConnectionManager;
-import com.intellij.database.dataSource.LocalDataSource;
 import com.intellij.database.model.DasObject;
 import com.intellij.database.psi.DbDataSource;
 import com.intellij.database.psi.DbElement;
 import com.intellij.database.psi.DbNamespaceImpl;
+import com.intellij.database.util.DbImplUtil;
 import com.intellij.database.util.GuardedRef;
 import com.intellij.openapi.progress.ProcessCanceledException;
 import com.intellij.openapi.progress.ProgressIndicator;
@@ -28,6 +28,7 @@
 import java.sql.Connection;
 import java.sql.SQLException;
 import java.util.List;
+import java.util.Objects;
 import java.util.Set;
 import java.util.stream.Collectors;
 
@@ -122,7 +123,7 @@ public String generateDiff(DbDataSource target, DbDataSource reference) {
 
     private GuardedRef<DatabaseConnection> acquireConnection(DbDataSource dataSource) throws SQLException {
         return DatabaseConnectionManager.getInstance()
-                .build(this.project, (LocalDataSource) dataSource.getDelegate()).create();
+                .build(this.project, Objects.requireNonNull(DbImplUtil.getMaybeLocalDataSource(dataSource))).create();
     }
 
     private Database getDatabase(GuardedRef<DatabaseConnection> connectionRef) throws DatabaseException {