Bug/fix imports