add custom wait_for time. Fixes #129