fix #13

@Goldenstone check this pr.