Updated after-stage rule for modified directory structure