Small imports bug fix