Decoration: Add support for zero probability, fix breakage from last commit