Adding implicit waits for more deterministic integration tests