ModuleNotFoundError: No module named 'kaldi_data_generator.image_utils'; 'kaldi_data_generator' is not a package

Apparently that's because the package is called kaldi_data_generator and then the script is called kaldi_data_generator.py, it works if I rename it like kaldi_data_generator_0.py