Oops - forgot to update a variable.