Fix imports for hacking