#2 fix bug