make colorization a little bit more robust