attempt to fix #5

Should be fine, judging by highlight/shell#L226
Also, tested locally, seems to work fine.