Updated api, should work now