diff --git a/hw/rtl/core/VX_execute.sv b/hw/rtl/core/VX_execute.sv
index 86e3d9ed..5077bd25 100644
--- a/hw/rtl/core/VX_execute.sv
+++ b/hw/rtl/core/VX_execute.sv
@@ -418,7 +418,7 @@ module VX_execute import VX_gpu_pkg::*; #(
         input [31:0] denom_bits;
         begin
         `ifdef SV_DPI
-            scalar_tmem_softmax_prob_pack = 32'(dpi_softmax_prob_to_fp8e4m3x4(1'b1, int'(score_bits), int'(max_bits), int'(denom_bits)));
+            scalar_tmem_softmax_prob_pack = 32'(dpi_softmax_prob_to_f16x2(1'b1, int'(score_bits), int'(max_bits), int'(denom_bits)));
         `else
             scalar_tmem_softmax_prob_pack = '0;
         `endif