Fix up some code