Fixed minor code