diff --git a/samples/float_model_embedding/barcode.c b/samples/float_model_embedding/barcode.c
index 31c2d86..b49bea8 100644
--- a/samples/float_model_embedding/barcode.c
+++ b/samples/float_model_embedding/barcode.c
@@ -35,7 +35,7 @@
 }
 
 iree_status_t load_input_data(const MlModel *model, void **buffer) {
-  iree_status_t result = alloc_input_data(model, buffer);
+  iree_status_t result = alloc_input_buffer(model, buffer);
   // Populate initial value
   srand(768954);
   if (iree_status_is_ok(result)) {
diff --git a/samples/quant_model_embedding/scenenet_v2.c b/samples/quant_model_embedding/scenenet_v2.c
index 16ccfd5..b197496 100644
--- a/samples/quant_model_embedding/scenenet_v2.c
+++ b/samples/quant_model_embedding/scenenet_v2.c
@@ -37,9 +37,9 @@
   // Populate initial value
   srand(55555555);
   if (iree_status_is_ok(result)) {
-  for (int i = 0; i < model->input_length[0]; ++i) {
-    ((uint8_t *)*buffer)[i] = (uint8_t)rand();
-  }
+    for (int i = 0; i < model->input_length[0]; ++i) {
+      ((uint8_t *)*buffer)[i] = (uint8_t)rand();
+    }
   }
   return result;
 }