diff --git a/server/predict.py b/server/predict.py
index fced9d4..edf7fac 100644
--- a/server/predict.py
+++ b/server/predict.py
@@ -49,9 +49,9 @@ def generate_midi(midi_data, total_seconds=10):
             1, qpm)
     generate_section.end_time_seconds = total_seconds
     # generate_response = generator_map[generator_name].generate(generate_request)
-    generate_response = basic_generator.generate(generate_request.input_sequence, generate_request.generator_options)
+    generate_response = basic_generator.generate(generate_request)
     output = tempfile.NamedTemporaryFile()
     midi_io.sequence_proto_to_midi_file(
-          generate_response, output.name)
+          generate_response.generated_sequence, output.name)
     output.seek(0)
-    return output
\ No newline at end of file
+    return output