fixing a bug from the code import