Fix stale import