ML API bugfix

Signed-off-by: Christoph M. Wintersteiger <cwinter@microsoft.com>