fix bug

#1