Fix test (remove n_features from init)