Updating tests for new custom close button logic