Fix ssa_forward.py. Port to 0.6.