Import bug fix