Fix for issue https://github.com/HouseTrip/cms-puppet/issues/1

I had modified code for issue https://github.com/HouseTrip/cms-puppet/issues/1 . Can you merge it ?