diff --git a/src/main.rs b/src/main.rs
index 8e4eb1b..a20292b 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -35,7 +35,7 @@ fn pretty_retrieve(id: PasteId) -> Option<Template> {
     map.insert("code", contents);
     let rendered = Template::render("pretty", &map);
 
-    match tree_magic::from_filepath(filepath).contains("text") {
+    match tree_magic::match_filepath("text/plain", filepath) {
         true    =>  Some(rendered),
         false   =>  None
     }