Type hint bugfix