Automated test edits
Fix for the automated test