Fix Marginalize example