Fixes #21 issue.

My first contribution :)