Improve the ImportHook class