Tests pass.

Had to modify the version to be generation instead.