minor: fixing import