more api fix
Tiny change