in the provided case, you must reinclude it at test time, but not binary time, no?