diff --git a/src/aiidalab_qe/app/main.py b/src/aiidalab_qe/app/main.py
index afe46af72..56ca038ce 100644
--- a/src/aiidalab_qe/app/main.py
+++ b/src/aiidalab_qe/app/main.py
@@ -18,7 +18,7 @@
 from aiidalab_qe.app.submission import SubmitQeAppWorkChainStep
 from aiidalab_qe.app.submission.model import SubmissionModel
 from aiidalab_qe.common.widgets import LoadingWidget
-from aiidalab_widgets_base import WizardAppWidget, WizardAppWidgetStep
+from aiidalab_widgets_base import WizardAppWidget
 
 
 class App(ipw.VBox):
diff --git a/src/aiidalab_qe/common/widgets.py b/src/aiidalab_qe/common/widgets.py
index a27152cf3..2c511c2ad 100644
--- a/src/aiidalab_qe/common/widgets.py
+++ b/src/aiidalab_qe/common/widgets.py
@@ -980,7 +980,7 @@ def _on_render_click(self, _):
 class LazyLoadedOptimade(LazyLoadedStructureImporter):
     warning_message = "OPTIMADE may take some time to load"
 
-    def _get_children(self):
+    def _get_widget(self):
         from aiidalab_widgets_base.databases import OptimadeQueryWidget
 
         return OptimadeQueryWidget(embedded=False)