FIX: typo in stage2.m