fix bug for import testing