Fixes #4. Thanks @shedar!