Typescript fixes