Fixed issue from code review.