Fix code issue base on Dale feedback.

here ya go