Fixed bug in find method