diff --git a/flax/subs/frontwhite b/flax/subs/frontwhite
--- a/flax/subs/frontwhite
+++ b/flax/subs/frontwhite
@@ -1,11 +1,11 @@
-1 : 1
-2 : 1
-3 : 1
-4 : 1
-5 : 1
-7 : 1
-8 : 1
-9 : 1
-10 : 1
-11 : 1
-12 : 0.5
+2 : 1.0
+diag : 1.0
+diag-back : 1.0
+f1 : 1.0
+f1.5 : 1.0
+f2 : 1.0
+f3 : 1.0
+f4 : 1.0
+f5 : 1.0
+f6 : 1.0
+f7 : 0.5