For anyone who has got this issue. I got it to work by passing it like this

Having the template data key and then sending the parameters one by one

$templateData = [];
                $templateData = [
                    'orderNumber' => $order->getOrderNumber(),
                ];
                $this->mailService->send($dataBag->all(), $context, $templateData);