- fixed a bug with unit import