fixed a bug with import