Type hint bug fix