Fixes to FeatureSelection Restart Procedure