Fix imports bug