Also updated the interface to prevent compiler warning.
Fixes static analyser issue.
Merge pull request #4 from jeroenleenarts/patch-2

Thanks Jeroen. A little TLC on the code is good.