fix ts bug