About VQ-VAE

Thanks to your great implementation! I have a question about VQ-VAE. The paper of VQ VAE says that argmin of Ze and ej is measured using L2 distance. You implementation actually calculate their distance but didn't using mse. I am curious about the reason~ Thanks~~~