Fixed #116

Thanks Freshmintyy!