I have fixed this by doing a try/except on the import of the function, but we need to remove this completely because I'm sure they'll break it again.