Fix thes stubs dep in augment