Bug fix for error found in review.