Ajay | #0000 | Added wait to test