fix #11

Thanks @ZDroid