fixed issue in first commit