Fixed #20. Thanks Phoenixf129