Fixed bug in import