Add a test to check the status code