Fix missing set_duration() refactor