fix feature_wrapper bugs