diff --git a/src/bin/protoc-gen-roto.rs b/src/bin/protoc-gen-roto.rs
index b06fb59..f0f111c 100644
--- a/src/bin/protoc-gen-roto.rs
+++ b/src/bin/protoc-gen-roto.rs
@@ -1,10 +1,9 @@
 use env_logger::init;
 use log::{error, info};
 use roto::generator::generate_rust_code;
-use roto::google::protobuf::descriptor::{
-    FileDescriptorSet
+use roto::proto_gen::google::protobuf::descriptor::{
+    CodeGeneratorRequest, CodeGeneratorResponse, FileDescriptorSet, ResponseFile,
 };
-use roto::google::protobuf::compiler::plugin::{CodeGeneratorRequest, CodeGeneratorResponse};
 use roto::ProtoBuilder;
 use std::io::{self, Read, Write};
 
diff --git a/src/generator.rs b/src/generator.rs
index 7b36d4d..249b9e7 100644
--- a/src/generator.rs
+++ b/src/generator.rs
@@ -1,8 +1,8 @@
 
-use crate::google::protobuf::descriptor::{
-    FileDescriptorSet, FieldDescriptorProto, DescriptorProto, EnumDescriptorProto, FileDescriptorProto
+
+use crate::proto_gen::google::protobuf::descriptor::{
+    CodeGeneratorRequest, CodeGeneratorResponse, FileDescriptorSet, ResponseFile, DescriptorProto, EnumDescriptorProto, FileDescriptorProto, FieldDescriptorProto
 };
-use crate::google::protobuf::compiler::plugin::{CodeGeneratorRequest, CodeGeneratorResponse};
 use crate::{ProtoAccessor, Result, RotoError};
 use std::str;
 use std::collections::{HashMap, HashSet};
diff --git a/src/lib.rs b/src/lib.rs
index 0bb962f..5f18759 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,6 +1,6 @@
 pub mod proto_gen;
 pub mod generator;
-pub mod google;
+// pub mod google;
 // Uncomment this to check if the code compiles
 // #[path = "../proto/google/protobuf/descriptor.rs"]
 // pub mod descriptor;
diff --git a/tests/test_nested_protos.rs b/tests/test_nested_protos.rs
new file mode 100644
index 0000000..59c4147
--- /dev/null
+++ b/tests/test_nested_protos.rs
@@ -0,0 +1,44 @@
+use roto::generator::generate_rust_code;
+use roto::proto_gen::google::protobuf::descriptor::FileDescriptorSet;
+use std::fs;
+
+#[test]
+fn test_nested_proto_generation_contains_modules() {
+    let request_path = "data/request.bin";
+    if !std::path::Path::new(request_path).exists() {
+        panic!("data/request.bin not found. This test requires the sample request binary.");
+    }
+
+    let data = fs::read(request_path).expect("Failed to read request.bin");
+
+    // The existing test logic to build a FileDescriptorSet from CodeGeneratorRequest
+    // We can simplify this by just wrapping the data if it's already a FileDescriptorSet,
+    // but request.bin is usually a CodeGeneratorRequest.
+
+    // Let's use the same logic as build_generated_code.rs to get a FileDescriptorSet
+    let request = roto::proto_gen::google::protobuf::compiler::plugin::CodeGeneratorRequest::new(&data)
+        .expect("Failed to parse CodeGeneratorRequest");
+
+    let mut set_buf = Vec::new();
+    for file_res in request.proto_file() {
+        let (file_data, _) = file_res.expect("Failed to iterate proto_file");
+        set_buf.push(10);
+        let len = file_data.len() as u64;
+        let mut len_buf = [0u8; 10];
+        let len_size = roto::write_varint(len, &mut len_buf).expect("Failed to write varint length");
+        set_buf.extend_from_slice(&len_buf[..len_size]);
+        set_buf.extend_from_slice(file_data);
+    }
+    let set = FileDescriptorSet::new(&set_buf).expect("Failed to create FileDescriptorSet");
+
+    let generated_files = generate_rust_code(&set, None, false);
+
+    let all_code: String = generated_files.into_iter().map(|(_, content)| content).collect();
+    println!("Generated Code:\n{}", all_code);
+
+    // We want to see if any message has a nested module.
+    // Since we don't know exactly what's in request.bin, we'll look for ANY 'pub mod' inside the generated code
+    // that isn't at the top level (though the generator puts them inside the message definition).
+    assert!(all_code.contains("pub mod "), "Generated code should contain at least one nested module for nested types");
+    assert!(all_code.contains("pub struct "), "Generated code should contain structs");
+}