diff --git a/CoreMLDemo/ViewController.swift b/CoreMLDemo/ViewController.swift
index a4f3e23..3d7058e 100644
--- a/CoreMLDemo/ViewController.swift
+++ b/CoreMLDemo/ViewController.swift
@@ -59,11 +59,11 @@ extension ViewController: UIImagePickerControllerDelegate {
         dismiss(animated: true, completion: nil)
     }
     
-    func imagePickerController(_ picker: UIImagePickerController, didFinishPickingMediaWithInfo info: [String : Any]) {
+    func imagePickerController(_ picker: UIImagePickerController, didFinishPickingMediaWithInfo info: [UIImagePickerController.InfoKey : Any]) {
         
         picker.dismiss(animated: true)
         classifier.text = "Analyzing Image..."
-        guard let image = info["UIImagePickerControllerOriginalImage"] as? UIImage else {
+        guard let image = info[UIImagePickerController.InfoKey.originalImage] as? UIImage else {
             return
         } //1