Fix up equipment add function