Add mock patch