Reflect changes in tests