Update type annotation