bug fix

v1.1 bug fix for prodution