diff --git a/.emacs.d/init.el b/.emacs.d/init.el
index e2cbaa1..c1618f1 100644
--- a/.emacs.d/init.el
+++ b/.emacs.d/init.el
@@ -161,13 +161,21 @@
 
 ;; Docker
 (use-package dockerfile-mode
-  :defer t)
+   :ensure t
+   :defer t)
 
 ;; YAML
 (use-package yaml-mode
-  :mode "\\.yml\\'")
-(add-hook 'yaml-mode-hook 'flymake-yaml-load)
-
+  :ensure t
+  :mode
+  (
+   ("\\.yml\\'" . yaml-mode)
+   )
+  :hook
+  (
+   (yaml-mode-hook . flymake-yaml-load)
+    )
+ )
 ;; Git
 (global-set-key (kbd "C-x g") 'magit-status)