Fixed small update bugs