diff --git a/meshroom/ui/commands.py b/meshroom/ui/commands.py
index 9c55cbf8..a87e01d4 100644
--- a/meshroom/ui/commands.py
+++ b/meshroom/ui/commands.py
@@ -1,6 +1,6 @@
 from PySide2.QtWidgets import QUndoCommand, QUndoStack
 from PySide2.QtCore import Property, Signal
-from meshroom.core.graph import Node
+from meshroom.core.graph import Node, ListAttribute
 
 
 class UndoCommand(QUndoCommand):