Fixes #11 breaking change from release 0.3.1.