Update checkout controller override

The checkout controller decorator was a little out of date. I think this fixes #33.