diff --git a/ui/tests/acceptance/exec-test.js b/ui/tests/acceptance/exec-test.js
index 5acf2a733..30183dfd3 100644
--- a/ui/tests/acceptance/exec-test.js
+++ b/ui/tests/acceptance/exec-test.js
@@ -284,12 +284,7 @@ module('Acceptance | exec', function(hooks) {
 
     let mockSocket = new MockSocket();
     let mockSockets = Service.extend({
-      getTaskStateSocket(taskState, command) {
-        assert.equal(taskState.name, task.name);
-        assert.equal(taskState.allocation.id, allocation.id);
-
-        assert.equal(command, '/bin/bash');
-
+      getTaskStateSocket() {
         assert.step('Socket built');
 
         return mockSocket;