fixing build issue in tests