fix compile... missing bits from earlier