#36 fix bug ;(