Replace Timeout with something else

### 🤔 What's the problem you've observed?

We use `Timeout` which is considered dangerous.

### ✨ Do you have a proposal for making it better?

Use a loop with a test of how much time has passes since we started looping

### 📚 Any additional context?

TODO: I should probably look up why `Timeout` is considered dangerous.