Updated due to API changes.

Signed-off-by: Man_cub <mcmancub@gmail.com>