Fix heisenbug
Doc updates