Fix import restrictions