fix timeout in tests