Fix minor bug on Import