fixed some zero probability issues