X-Git-Url: http://git.iain.cx/?a=blobdiff_plain;f=state%2Fstate.go;h=362ec100812ae4d9513637c3030d34d60347b407;hb=586d58db4a534800775426446503428a629044fe;hp=33a86ea89cf5ee511e15824c995ed2e23c54ea93;hpb=a67f18670b644d854db829e71d125ae191f1dfdf;p=pstop.git

diff --git a/state/state.go b/state/state.go
index 33a86ea..362ec10 100644
--- a/state/state.go
+++ b/state/state.go
@@ -63,6 +63,8 @@ func (state *State) Setup(dbh *sql.DB) {
 	state.screen.Initialise()
 
 	state.setup_instruments.EnableMutexMonitoring(dbh)
+	state.setup_instruments.EnableStageMonitoring(dbh)
+
 	_, variables := lib.SelectAllGlobalVariablesByVariableName(state.dbh)
 	// setup to their initial types/values
 	state.fsbi = fsbi.NewFileSummaryByInstance(variables)
@@ -481,7 +483,7 @@ func (state *State) ScreenSetSize(width, height int) {
 func (state *State) Cleanup() {
 	state.screen.Close()
 	if state.dbh != nil {
-		state.setup_instruments.Restore(state.dbh)
+		state.setup_instruments.RestoreConfiguration(state.dbh)
 		_ = state.dbh.Close()
 	}
 }