diff --git a/aimodel/src/lib/ai/components/LayerContrastiveEncoder.py b/aimodel/src/lib/ai/components/LayerContrastiveEncoder.py
index 524eff4..c74232e 100644
--- a/aimodel/src/lib/ai/components/LayerContrastiveEncoder.py
+++ b/aimodel/src/lib/ai/components/LayerContrastiveEncoder.py
@@ -45,7 +45,7 @@ class LayerContrastiveEncoder(tf.keras.layers.Layer):
 		# self.embedding = tf.keras.layers.Dense(self.param_feature_dim)
 		
 		if summary_file:
-			summarywriter(self.encoder, append=True)
+			summarywriter(self.encoder, summary_file, append=True)
 	
 	def get_config(self):
 		config = super(LayerContrastiveEncoder, self).get_config()