Setting the correct variable