diff --git a/sources/qetdiagrameditor.cpp b/sources/qetdiagrameditor.cpp
index 580b42606..72616258f 100644
--- a/sources/qetdiagrameditor.cpp
+++ b/sources/qetdiagrameditor.cpp
@@ -1128,17 +1128,25 @@ void QETDiagramEditor::slot_rotateTexts() {
 }
 
 /**
-	Effectue l'action "mode selection" sur le schema en cours
-*/
-void QETDiagramEditor::slot_setSelectionMode() {
-	if(currentDiagram()) currentDiagram() -> setSelectionMode();
+ * @brief QETDiagramEditor::slot_setSelectionMode
+ * Set all diagram view of the current project to selection mode
+ */
+void QETDiagramEditor::slot_setSelectionMode()
+{
+	if (ProjectView *pv = currentProject())
+		foreach(DiagramView *dv, pv -> diagrams())
+			dv -> setSelectionMode();
 }
 
 /**
-	Effectue l'action "mode visualisation" sur le schema en cours
-*/
-void QETDiagramEditor::slot_setVisualisationMode() {
-	if(currentDiagram()) currentDiagram() -> setVisualisationMode();
+ * @brief QETDiagramEditor::slot_setVisualisationMode
+ * Set all diagram view of the current project to visualisation mode
+ */
+void QETDiagramEditor::slot_setVisualisationMode()
+{
+	if (ProjectView *pv = currentProject())
+		foreach(DiagramView *dv, pv -> diagrams())
+			dv -> setVisualisationMode();
 }
 
 /**
diff --git a/sources/qetdiagrameditor.h b/sources/qetdiagrameditor.h
index f516dfc0c..867bee2b6 100644
--- a/sources/qetdiagrameditor.h
+++ b/sources/qetdiagrameditor.h
@@ -52,18 +52,16 @@ class QETDiagramEditor : public QETMainWindow {
 		void setUpToolBar       ();
 		void setUpMenu          ();
 	
-	// methods
+		// methods
 	public:
-	void closeEvent(QCloseEvent *);
-	QList<ProjectView *> openedProjects() const;
-	void addProjectView(ProjectView *);
-	bool openAndAddProject(const QString &, bool = true, bool = true);
-	QList<DiagramView *> projectViews() const;
-	QList<QString> editedFiles() const;
-	ProjectView *viewForFile(const QString &) const;
+		void                 closeEvent        (QCloseEvent *);
+		QList<ProjectView *> openedProjects    () const;
+		void                 addProjectView    (ProjectView *);
+		bool                 openAndAddProject (const QString &, bool = true, bool = true);
+		QList<QString>       editedFiles       () const;
+		ProjectView         *viewForFile       (const QString &) const;
 	
 	protected:
-	//void actions();
 	virtual bool event(QEvent *);
 	
 	private: