Fix up code.