Improve the judgment of src file. Fix #66