Switched to using find rather than equality