Working to fix update