Update to work with current 2048