Fixed wait-for declaration error in mock impl