Fixing tests with timeout.