Use estraverse.traverse instead of estraverse.replace