Fixed, just forgot to set generate prob