Fixes issue 9

Fixes #9.