[ML][Single] Fix the crash issue in single API