Fix bug

Oops - bug - removed a variable, but left a reference