from litex.soc.tools.remote.comm_uart import CommUART
-from litex.soc.tools.remote.server import RemoteServer
-from litex.soc.tools.remote.client import RemoteClient
+from litex.soc.tools.remote.comm_udp import CommUDP
+from litex.soc.tools.remote.comm_pcie import CommPCIe
+from litex.soc.tools.remote.litex_server import RemoteServer
+from litex.soc.tools.remote.litex_client import RemoteClient