Corrected mistake in expected value calculation