Fixed friend type issue