WIP on handling train & test with different enums