diff --git a/src/Loggers/LogCustomScalar.jl b/src/Loggers/LogCustomScalar.jl
index 4103a35..c2b42f5 100644
--- a/src/Loggers/LogCustomScalar.jl
+++ b/src/Loggers/LogCustomScalar.jl
@@ -1,3 +1,6 @@
+using .tensorboard_plugin_custom_scalar
+using .tensorboard_plugin_custom_scalar: var"MarginChartContent.Series" as MarginChartContent_Series
+
 # possible chart types
 @enum tb_chart_type tb_multiline=1 tb_margin=2