Fix to use the new API