Fix and update

Update the import declaration