Update test after introducing enum to match previous version.