diff --git a/src/repository/dockerhub.rs b/src/repository/dockerhub.rs
index 373b874..e1fe087 100644
--- a/src/repository/dockerhub.rs
+++ b/src/repository/dockerhub.rs
@@ -88,13 +88,16 @@ impl DockerHub {
 
 #[cfg(test)]
 mod tests {
-    use crate::repository::dockerhub::Repo;
+    use crate::repository::dockerhub::DockerHub;
     #[test]
     fn test_check_repo() {
-        assert_eq!(Repo::check_repo("nginx").unwrap(), "library/nginx");
-        assert_eq!(Repo::check_repo("library/nginx").unwrap(), "library/nginx");
+        assert_eq!(DockerHub::check_repo("nginx").unwrap(), "library/nginx");
         assert_eq!(
-            Repo::check_repo("rocketchat/rocket.chat").unwrap(),
+            DockerHub::check_repo("library/nginx").unwrap(),
+            "library/nginx"
+        );
+        assert_eq!(
+            DockerHub::check_repo("rocketchat/rocket.chat").unwrap(),
             "rocketchat/rocket.chat"
         );
     }