Adding theta output correctly to unfolding