Update to work version