Fix unnecessary variable left over from refactor