Fix parsing single-level probability lists