Fix incorrect return type
Use alias type name