Fixed minor issue with variable name