restored total prob instead of average