diff --git a/flybody/agents/ray_distributed_dmpo.py b/flybody/agents/ray_distributed_dmpo.py
index 35f62ac..609687e 100755
--- a/flybody/agents/ray_distributed_dmpo.py
+++ b/flybody/agents/ray_distributed_dmpo.py
@@ -1,6 +1,6 @@
 """Classes for DMPO agent distributed with Ray."""
 
-from typing import Optional, Iterator, Callable, Any
+from typing import Optional, Iterator, Callable
 import socket
 import dataclasses
 import copy