Fixing import & bumping