v2.1 fix bug