Fixed minor variable issue