fixing dimenson in screenToWorld

Author: perminder-17

src/core/environment.js

@@ -1314,25 +1314,24 @@ function environment(p5, fn, lifecycles){
    *
    */
   fn.screenToWorld = function(screenPosition) {
+    // remember how many components the caller actually provided to createVector
+    const origDimension = this.dimension;
+  
     if (typeof screenPosition === 'number') {
       // We got passed numbers, convert to vector
       screenPosition = this.createVector(...arguments);
     }
-
+  
     const matrix = this._renderer.getWorldToScreenMatrix();
-
-    if (screenPosition.dimensions === 2) {
+    if (origDimension === 2) {
       // Calculate a sensible Z value for the current camera projection that
       // will result in 0 once converted to world coordinates
       let z = matrix.mat4[14] / matrix.mat4[15];
       screenPosition = this.createVector(screenPosition.x, screenPosition.y, z);
     }
 
     const matrixInverse = matrix.invert(matrix);
-
-    const worldPosition = matrixInverse
-      .multiplyAndNormalizePoint(screenPosition);
-    return worldPosition;
+    return matrixInverse.multiplyAndNormalizePoint(screenPosition);
   };
 
   /**