fixed so that regression testing works