Its not safe to change the existing tests. Instead of that you can add new field to entity.