Fixed issue 14: thanks Yang