#8 fix bug