Change the interface

We want to use the WaitTime() (time.Duration) interface because it's a
bit cleaner.