Fixed minor type issue