fix #7 :)