Fix code import