update to work with beta 3