Fixed ts error