diff --git a/webgoat-lessons/path-traversal/src/test/java/org/owasp/webgoat/path_traversal/ProfileUploadRetrievalTest.java b/webgoat-lessons/path-traversal/src/test/java/org/owasp/webgoat/path_traversal/ProfileUploadRetrievalTest.java
index 1e1f41382..bbade5855 100644
--- a/webgoat-lessons/path-traversal/src/test/java/org/owasp/webgoat/path_traversal/ProfileUploadRetrievalTest.java
+++ b/webgoat-lessons/path-traversal/src/test/java/org/owasp/webgoat/path_traversal/ProfileUploadRetrievalTest.java
@@ -75,8 +75,7 @@ public class ProfileUploadRetrievalTest extends LessonTest {
     @Test
     public void unknownFileShouldGiveDirectoryContents() throws Exception {
         mockMvc.perform(get("/PathTraversal/random-picture?id=test"))
-                .andExpect(status().is(200))
-                .andExpect(content().string(containsString("cats/8.jpg")))
-                .andExpect(content().contentTypeCompatibleWith(MediaType.IMAGE_JPEG));
+                .andExpect(status().is(404))
+                .andExpect(content().string(containsString("cats/8.jpg")));
     }
 }
\ No newline at end of file