Fix internal type issue