Add missing import for eltype