Addressed bug found in code review comment.