Issue 16 fix

Created by: TheSageMarmot

I tested in Ubuntu 18.04. Author of #16 (closed) tested in Ubuntu 20.04. Should this be tested in other distro's besides Ubuntu?