1 files changed, 6 insertions, 0 deletions

diff --git a/src/viz.c b/src/viz.c
index 8e31387..ebe1a4c 100644
--- a/src/viz.c
+++ b/src/viz.c
@@ -272,6 +272,7 @@ static const char* basm_op_str[] = {
     [OP_JMP_LT]     = "jlt   ",
     [OP_JMP_GE]     = "jge   ",
     [OP_JMP_LE]     = "jle   ",
+    [OP_RETURN]     = "ret   ",
 };
 
 void
@@ -311,11 +312,16 @@ viz_instruction(Instruction *inst, LineInfo *line) {
                 viz_operand(inst->src_b);
             }
         } break;
+        case OP_RETURN:
         case OP_JMP:
         case OP_LABEL: {
             printf(" ");
             viz_operand(inst->dst);
         } break;
+        case OP_CP8:
+        case OP_CP16:
+        case OP_CP32:
+        case OP_CP64:
         case OP_JMP_TRUE:
         case OP_JMP_FALSE: {
             printf(" ");