Fix on dumb stuff I left