Fixed a bug while importing