fix timeout in test