added missing squaring