Fix #2

Thanks to @gawbul