Fix call to add_features
Fixes