diff --git a/src/app/webide/webide.component.html b/src/app/webide/webide.component.html
index 9ae5301..abee9d4 100644
--- a/src/app/webide/webide.component.html
+++ b/src/app/webide/webide.component.html
@@ -9,7 +9,7 @@
   </button>
 </div>
 
-<div (keypress)="setSaveButtonStateDirty(); setDeployButtonStateTodeploy(); resetAutoSaveCountdown()"
+<div (keypress)="setSaveButtonState('DIRTY'); setDeployButtonState('TODEPLOY'); resetAutoSaveCountdown()"
   ace-editor
   [(text)]="code"
   [mode]="language"
diff --git a/src/app/webide/webide.component.ts b/src/app/webide/webide.component.ts
index 3de38fc..6a6498b 100644
--- a/src/app/webide/webide.component.ts
+++ b/src/app/webide/webide.component.ts
@@ -50,7 +50,7 @@ export class WebideComponent implements OnInit {
     this.subscribeWS();
     this.requestCode();
     this.processManagerService.init();
-    this.processManagerService.subscribeCallback('login', (event) => { this.setDeployButtonStateDeployed(); });
+    this.processManagerService.subscribeCallback('login', (event) => { this.setDeployButtonState('DEPLOYED'); });
     this.resetAutoSaveCountdown();
   }
 
@@ -100,16 +100,12 @@ export class WebideComponent implements OnInit {
     this.requestCode();
   }
 
-  setSaveButtonStateDirty() {
-    this.saveButtonState = 'DIRTY';
+  setSaveButtonState(state: string) {
+    this.saveButtonState = state;
   }
 
-  setDeployButtonStateDeployed() {
-    this.deployButtonState = 'DEPLOYED';
-  }
-
-  setDeployButtonStateTodeploy() {
-    this.deployButtonState = 'TODEPLOY';
+  setDeployButtonState(state: string) {
+    this.deployButtonState = state;
   }
 
   deployCode() {