And fixed another small error