ML API: Undoing earlier changes.

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