fix a missing function bug

[git-p4: depot-paths = "//depot/private/jchin/HBAR-DTK/": change = 121987]
fix missing function