Fix slow tests