Code fixups