Fix error on add function