minor update on bug fix