diff --git a/ML/Pytorch/CNN_architectures/pytorch_inceptionet.py b/ML/Pytorch/CNN_architectures/pytorch_inceptionet.py
index 38faef9..bdc2057 100644
--- a/ML/Pytorch/CNN_architectures/pytorch_inceptionet.py
+++ b/ML/Pytorch/CNN_architectures/pytorch_inceptionet.py
@@ -11,7 +11,7 @@ Programmed by Aladdin Persson <aladdin.persson at hotmail dot com>
 
 # Imports
 import torch
-import torch.nn as nn  # All neural network modules, nn.Linear, nn.Conv2d, BatchNorm, Loss functions
+from torch import nn
 
 
 class GoogLeNet(nn.Module):