Widget test incosistencies

Work in progress

- Use waitFor instead of sleeping.