minor modification for import