Bug fix

Fixes #28 Thanks to @crapzDK for reporting this.