Fixed some selenium wait problems