Get discrete_law test working again