Fix another incorrect import