small fix with import