fixing import after the renaming