Fix missing mock tweak
Fix spec