fix: Ensure segment dir is in-scope during proving

fix: Ensure segment dir is in-scope during proving #418