Update import from BTKInjector