Fix a type issue