compile fix, missing bits from the last commit.