fix issue with import