solution for the tmp files cleanup

fix imports