Fix issues with site-edit.php