V0.2 fix bug