Diffusion Craft 8e6d95bd8afa

fix import error

Authored by sengels.