Fix errors mike found