fix: Move necessary import out of type checking block

fix: Move necessary import out of type checking block #1974