FIX: wrong input dir in stage 3