Fix import design