Type hint fix