puppet_agent : Ensuring package version fails on Amazon

Basic Info
Module Version: Current git master
Puppet Version: 5.1.0
OS Name/Version: CentOS 7 (master), Amazon Linux 2017.03 (client)

Describe your issue in as much detail as possible...
When setting puppet_agent::package_version to an explicit version (e.g., "5.1.0", puppet runs fail

Desired Behavior:
The package is ensured to the correct version

Actual Behavior:
Puppet fails with an error (see below):

Error: Could not update: Failed to update to version 5.1.0-1.el2017, got version 5.1.0-1.el6 instead

Error: /Stage[main]/Puppet_agent::Install/Package[puppet-agent]/ensure: change from '5.1.0-1.el6' to '5.1.0-1.el2017' failed: Could not update: Failed to update to version 5.1.0-1.el2017, got version 5.1.0-1.el6 instead

I've proposed a fix at:
https://github.com/puppetlabs/puppetlabs-puppet_agent/pull/249