diff --git a/src/lib.zig b/src/lib.zig
index 6e494a2..b380724 100644
--- a/src/lib.zig
+++ b/src/lib.zig
@@ -1,6 +1,7 @@
 const std = @import("std");
 
-pub const Tensor = @import("Tensor.zig").Tensor;
+pub const TensorStatic = @import("TensorStatic.zig").Tensor;
+pub const TensorAlloc = @import("TensorAlloc.zig").Tensor;
 pub const Dimensions = @import("Dimensions.zig");
 pub const Scales = @import("Scales.zig");
 pub const Base = @import("Base.zig");