small change to code in post