Add test for lazy mode