Update to use attribute