diff --git a/demos/web/websocket-server.py b/demos/web/websocket-server.py
index ec36808..9e0f921 100755
--- a/demos/web/websocket-server.py
+++ b/demos/web/websocket-server.py
@@ -61,7 +61,7 @@ parser.add_argument('--networkModel', type=str, help="Path to Torch network mode
                     default=os.path.join(openfaceModelDir, 'nn4.small2.v1.t7'))
 parser.add_argument('--imgDim', type=int,
                     help="Default image dimension.", default=96)
-parser.add_argument('--cuda', type=bool, default=False)
+parser.add_argument('--cuda', action='store_true')
 parser.add_argument('--unknown', type=bool, default=False,
                     help='Try to predict unknown people')
 parser.add_argument('--port', type=int, default=9000,