Fix counts to be keywords to work with new yaml library.