Updates to handle the new endpoint