fixed minor issue in prev PR