fix #231 :smile: