Fixed import error in tf_finder.py