Fix an incorrect import in the previous commit.