Fix choices argument in tests