Fixed augment dependency: updated to v4.0.1