Trying to fix drag updating bug