Fix compile error from previous commit (stale import)