Return types after import

src/components/EditorHeader/Modal/Modal.jsx

@@ -138,14 +138,18 @@ export default function Modal({
       return;
     }
 
-    const d = importSQL(ast, database === DB.GENERIC ? importDb : database, database);
+    const d = importSQL(
+      ast,
+      database === DB.GENERIC ? importDb : database,
+      database,
+    );
     if (importSource.overwrite) {
       setTables(d.tables);
       setRelationships(d.relationships);
       setTransform((prev) => ({ ...prev, pan: { x: 0, y: 0 } }));
       setNotes([]);
       setAreas([]);
-      setTypes([]);
+      setTypes(d.types ?? []);
       setUndoStack([]);
       setRedoStack([]);
     } else {

src/components/Workspace.jsx

@@ -355,7 +355,7 @@ export default function WorkSpace() {
         okButtonProps={{ disabled: selectedDb === "" }}
       >
         <div className="grid grid-cols-3 gap-4 place-content-center">
-          {databases.map((x) => (
+          {Object.values(databases).map((x) => (
             <div
               key={x.name}
               onClick={() => setSelectedDb(x.label)}