I added the check to the tests by adding a new page and also changing the method that clicks on the Save button. 
Please take a look and let me know what you think :)