Added some more code to hide the context of the previous test