diff --git a/test.ipynb b/test.ipynb
index 057b90b..6e054ba 100644
--- a/test.ipynb
+++ b/test.ipynb
@@ -519,7 +519,7 @@
         "        unique_colors = colors_sorted[unique_indices]\n",
         "\n",
         "        # Create color buffer and assign colors\n",
-        "        color_buffer = np.ones((img_size[1], img_size[0], 3), dtype=np.uint16)\n",
+        "        color_buffer = np.ones((img_size[1], img_size[0], 3), dtype=np.int16)\n",
         "        color_buffer = color_buffer*-1\n",
         "        color_buffer[unique_y, unique_x] = unique_colors\n",
         "\n",
@@ -531,7 +531,7 @@
         "        unique_colors = colors_sorted[unique_indices]\n",
         "\n",
         "        # Create color buffer and assign colors\n",
-        "        color_buffer = np.ones((img_size[1], img_size[0]), dtype=np.uint16)\n",
+        "        color_buffer = np.ones((img_size[1], img_size[0]), dtype=np.int16)\n",
         "        color_buffer = color_buffer*-1\n",
         "        color_buffer[unique_y, unique_x] = unique_colors\n",
         "\n",
@@ -545,7 +545,7 @@
         "        unique_colors = colors_sorted[unique_indices]\n",
         "\n",
         "        # Create color buffer and assign colors\n",
-        "        depth_buffer = np.zeros((img_size[1], img_size[0]), dtype=np.uint16)\n",
+        "        depth_buffer = np.zeros((img_size[1], img_size[0]), dtype=np.int16)\n",
         "        depth_buffer[unique_y, unique_x] = unique_colors\n",
         "\n",
         "        img_depth = depth_buffer\n",