That will be a safe approach. Watch out for the class imbalance in adding the error cases.