FIX: typo in stage4