Fixed a bug with the API