Fixes #1

Thank you @amarcadet for your suggestion and raising an issue.