correction in cited code

 an update according to  a bugfix in the original source