Fixed type issue