Fixing the test set-up for the change to personalize_elements