diff --git a/lerobot/common/policies/smolvla/smolvlm_with_expert.py b/lerobot/common/policies/smolvla/smolvlm_with_expert.py
index 891ff2091..ef369969d 100644
--- a/lerobot/common/policies/smolvla/smolvlm_with_expert.py
+++ b/lerobot/common/policies/smolvla/smolvlm_with_expert.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 import copy
-from typing import List, Optional, Union
+from typing import List, Optional
 
 import torch
 from torch import nn