Fixed a minor bug
Version bumped up to v4.0.1