Update from code review.  Fixed undefined variable