Fix some code

Nothing to see here.