Fixed another bad import