Fix another incorrect module import