Fix an incorrect import that sneaked in