Can we move this import into the class to make it an optional dependency? We already have quite a long list of deps.