Correctly modify the typescript :(