Fixing the edit page too