Fix variable name
Forgotten import