Fix import bugs