Fix another bad import