diff --git a/enformer_pytorch/modeling_enformer.py b/enformer_pytorch/modeling_enformer.py
index 48e8bd7..af587c0 100644
--- a/enformer_pytorch/modeling_enformer.py
+++ b/enformer_pytorch/modeling_enformer.py
@@ -81,7 +81,7 @@ def get_positional_features_gamma(positions, features, seq_len, stddev = None, s
     rate = mean / stddev ** 2
     probabilities = gamma_pdf(positions.float().abs()[..., None], concentration, rate)
     probabilities = probabilities + eps
-    outputs = probabilities / torch.amax(probabilities)
+    outputs = probabilities / torch.amax(probabilities, dim = -1, keepdim = True)
     return outputs
 
 def get_positional_embed(seq_len, feature_size, device):
diff --git a/setup.py b/setup.py
index 8478fdf..67dd287 100644
--- a/setup.py
+++ b/setup.py
@@ -4,11 +4,12 @@
   name = 'enformer-pytorch',
   packages = find_packages(exclude=[]),
   include_package_data = True,
-  version = '0.4.5',
+  version = '0.4.6',
   license='MIT',
   description = 'Enformer - Pytorch',
   author = 'Phil Wang',
   author_email = 'lucidrains@gmail.com',
+  long_description_content_type = 'text/markdown',
   url = 'https://github.com/lucidrains/enformer-pytorch',
   keywords = [
     'artificial intelligence',