From 8ec1566a272df896dc02e50ed9963b3276893ad0 Mon Sep 17 00:00:00 2001
From: Taro Kawazoe <zoeponta@gmail.com>
Date: Mon, 25 May 2020 12:36:52 +0900
Subject: [PATCH] Fix class names

---
 config.py                    | 2 +-
 diagrams/firebase/develop.py | 2 +-
 diagrams/firebase/grow.py    | 2 +-
 docs/nodes/firebase.md       | 4 ++--
 4 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/config.py b/config.py
index 5745d5b8..059b2942 100644
--- a/config.py
+++ b/config.py
@@ -52,7 +52,7 @@ UPPER_WORDS = {
     "aws": ("aws", "api", "ebs", "ec2", "efs", "emr", "rds", "ml", "mq", "nat", "vpc", "waf", "sdk"),
     "azure": ("ad", "b2c", "ai", "api", "cdn", "ddos", "dns", "fxt", "hana", "hd", "id", "sap", "sql", "vm"),
     "gcp": ("gcp", "ai", "api", "cdn", "dns", "gke", "gpu", "iap", "ml", "nat", "os", "sdk", "sql", "tpu", "vpn"),
-    "firebase": ("fcm", ),
+    "firebase": ("ab", "fcm", "ml"),
     "k8s": (
         "api", "cm", "ccm", "crb", "crd", "ds", "etcd", "hpa", "k8s", "ns", "psp", "pv", "pvc", "rb", "rs",
         "sa", "sc", "sts", "svc",
diff --git a/diagrams/firebase/develop.py b/diagrams/firebase/develop.py
index 2ff7e662..27426e51 100644
--- a/diagrams/firebase/develop.py
+++ b/diagrams/firebase/develop.py
@@ -24,7 +24,7 @@ class Hosting(_Develop):
     _icon = "hosting.png"
 
 
-class MlKit(_Develop):
+class MLKit(_Develop):
     _icon = "ml-kit.png"
 
 
diff --git a/diagrams/firebase/grow.py b/diagrams/firebase/grow.py
index 2a3cd8e8..b015dd3f 100644
--- a/diagrams/firebase/grow.py
+++ b/diagrams/firebase/grow.py
@@ -8,7 +8,7 @@ class _Grow(_Firebase):
     _icon_dir = "resources/firebase/grow"
 
 
-class AbTesting(_Grow):
+class ABTesting(_Grow):
     _icon = "ab-testing.png"
 
 
diff --git a/docs/nodes/firebase.md b/docs/nodes/firebase.md
index cfc416e8..f49a5c9c 100644
--- a/docs/nodes/firebase.md
+++ b/docs/nodes/firebase.md
@@ -15,7 +15,7 @@ Node classes list of firebase provider.
 - **diagrams.firebase.develop.Firestore**
 - **diagrams.firebase.develop.Functions**
 - **diagrams.firebase.develop.Hosting**
-- **diagrams.firebase.develop.MlKit**
+- **diagrams.firebase.develop.MLKit**
 - **diagrams.firebase.develop.RealtimeDatabase**
 - **diagrams.firebase.develop.Storage**
 
@@ -25,7 +25,7 @@ Node classes list of firebase provider.
 
 ## firebase.grow
 
-- **diagrams.firebase.grow.AbTesting**
+- **diagrams.firebase.grow.ABTesting**
 - **diagrams.firebase.grow.AppIndexing**
 - **diagrams.firebase.grow.DynamicLinks**
 - **diagrams.firebase.grow.InAppMessaging**