Fix import function