minor changes to get yaml working