fixing import bug