From bb6d06713f942fa23a560fe89bc25467ac630fcd Mon Sep 17 00:00:00 2001
From: Nanne Baars <nanne.baars@owasp.org>
Date: Mon, 9 Mar 2020 11:57:07 +0100
Subject: [PATCH] Fix failing test

---
 .../webgoat/path_traversal/ProfileUploadRetrievalTest.java   | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

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