Modifying test for discount model