diff --git a/bytebeat_compiler.py b/bytebeat_compiler.py
index 59d8eb3..1933ad6 100644
--- a/bytebeat_compiler.py
+++ b/bytebeat_compiler.py
@@ -90,7 +90,7 @@ if __name__ == "__main__":
 	makedirs(PATHS["build_dir"], exist_ok=True)
 
 	if not args.no_return:  # Insert return statement
-		bytebeat_contents = f"\treturn\n\n{bytebeat_contents}"
+		bytebeat_contents = f"return {bytebeat_contents}"
 
 	final_sample_rate_code = ""
 	if not args.final_sample_rate is None:
diff --git a/src/template.c b/src/template.c
index 3496a89..d88058f 100644
--- a/src/template.c
+++ b/src/template.c
@@ -84,8 +84,7 @@ bytebeat(long double w)
 {
 	`final_sample_rate_code`
 	uintmax_t t = (uintmax_t)w;
-	`bytebeat_contents`
-	;
+	`bytebeat_contents`;
 }
 
 #define USAGE "render_bytebeat [-s | --silent] [-d | --debug]"