Fix import

Update the import declaration