fix imports on mux