diff --git a/freqtrade/freqai/prediction_models/PyTorchClassifierClassifier.py b/freqtrade/freqai/prediction_models/PyTorchClassifier.py
similarity index 100%
rename from freqtrade/freqai/prediction_models/PyTorchClassifierClassifier.py
rename to freqtrade/freqai/prediction_models/PyTorchClassifier.py
diff --git a/freqtrade/freqai/prediction_models/MLPPyTorchClassifier.py b/freqtrade/freqai/prediction_models/PyTorchMLPClassifier.py
similarity index 95%
rename from freqtrade/freqai/prediction_models/MLPPyTorchClassifier.py
rename to freqtrade/freqai/prediction_models/PyTorchMLPClassifier.py
index 2f6705311..453995ce8 100644
--- a/freqtrade/freqai/prediction_models/MLPPyTorchClassifier.py
+++ b/freqtrade/freqai/prediction_models/PyTorchMLPClassifier.py
@@ -4,11 +4,11 @@ import torch
 
 from freqtrade.freqai.base_models.PyTorchModelTrainer import PyTorchModelTrainer
 from freqtrade.freqai.data_kitchen import FreqaiDataKitchen
-from freqtrade.freqai.prediction_models.PyTorchClassifierClassifier import PyTorchClassifier
+from freqtrade.freqai.prediction_models.PyTorchClassifier import PyTorchClassifier
 from freqtrade.freqai.prediction_models.PyTorchMLPModel import PyTorchMLPModel
 
 
-class MLPPyTorchClassifier(PyTorchClassifier):
+class PyTorchMLPClassifier(PyTorchClassifier):
     """
     This class implements the fit method of IFreqaiModel.
     int the fit method we initialize the model and trainer objects.
diff --git a/freqtrade/freqai/prediction_models/PyTorchMLPModel.py b/freqtrade/freqai/prediction_models/PyTorchMLPModel.py
index 0e6b3c7bb..f711a53a7 100644
--- a/freqtrade/freqai/prediction_models/PyTorchMLPModel.py
+++ b/freqtrade/freqai/prediction_models/PyTorchMLPModel.py
@@ -49,8 +49,8 @@ class PyTorchMLPModel(nn.Module):
         x = self.relu(self.input_layer(x))
         x = self.dropout(x)
         x = self.blocks(x)
-        logits = self.output_layer(x)
-        return logits
+        x = self.output_layer(x)
+        return x
 
 
 class Block(nn.Module):