Fixed a bug with the add