diff --git a/enhancer/models/model.py b/enhancer/models/model.py
index f8dd502..bdc49dd 100644
--- a/enhancer/models/model.py
+++ b/enhancer/models/model.py
@@ -3,7 +3,7 @@ from torch.optim import Adam
 import pytorch_lightning as pl
 
 from enhancer.data.dataset import Dataset
-from enhancer.utils.loss import LOSS_MAP, Avergeloss
+from enhancer.utils.loss import Avergeloss
 
 
 class Model(pl.LightningModule):
@@ -20,7 +20,7 @@ class Model(pl.LightningModule):
         assert num_channels ==1 , "Enhancer only support for mono channel models"
         self.save_hyperparameters("num_channels","sampling_rate","lr","loss")
         self.dataset = dataset
-
+        
     
     @property
     def dataset(self):
@@ -51,7 +51,7 @@ class Model(pl.LightningModule):
         return self.dataset.val_dataloader()
 
     def configure_optimizers(self):
-        return Adam(self.parameters, lr = self.hparams.lr)
+        return Adam(self.parameters(), lr = self.hparams.lr)
 
     def training_step(self,batch, batch_idx:int):