diff --git a/src/aiidalab_qe/app/structure/__init__.py b/src/aiidalab_qe/app/structure/__init__.py
index 2148b1a4b..95fd2908e 100644
--- a/src/aiidalab_qe/app/structure/__init__.py
+++ b/src/aiidalab_qe/app/structure/__init__.py
@@ -121,7 +121,7 @@ def render(self):
         )
 
         if structure:  # loaded from process
-            self.manager.viewer.structure = structure.get_ase()  # TODO why ase?
+            self.manager.viewer.structure = structure
 
         self.structure_name_text = ipw.Text(
             placeholder="[No structure selected]",