diff --git a/kmath-dimensions/src/commonMain/kotlin/scientifik/kmath/dimensions/Dimensions.kt b/kmath-dimensions/src/commonMain/kotlin/scientifik/kmath/dimensions/Dimensions.kt
index 87511b885..2eef69a0c 100644
--- a/kmath-dimensions/src/commonMain/kotlin/scientifik/kmath/dimensions/Dimensions.kt
+++ b/kmath-dimensions/src/commonMain/kotlin/scientifik/kmath/dimensions/Dimensions.kt
@@ -19,13 +19,11 @@ interface Dimension {
                 }
             }
         }
-
-        inline fun <reified D : Dimension> dim(): UInt {
-            return D::class.objectInstance?.dim ?: error("Dimension object must be a singleton")
-        }
     }
 }
 
+expect inline fun <reified D : Dimension>  Dimension.Companion.dim(): UInt
+
 object D1 : Dimension {
     override val dim: UInt = 1u
 }
diff --git a/kmath-dimensions/src/commonMain/kotlin/scientifik/kmath/dimensions/Wrappers.kt b/kmath-dimensions/src/commonMain/kotlin/scientifik/kmath/dimensions/Wrappers.kt
index 9985888b9..c78018753 100644
--- a/kmath-dimensions/src/commonMain/kotlin/scientifik/kmath/dimensions/Wrappers.kt
+++ b/kmath-dimensions/src/commonMain/kotlin/scientifik/kmath/dimensions/Wrappers.kt
@@ -66,7 +66,8 @@ interface DPoint<T, D : Dimension> : Point<T> {
 /**
  * Dimension-safe point wrapper
  */
-inline class DPointWrapper<T, D : Dimension>(val point: Point<T>) : DPoint<T, D> {
+inline class DPointWrapper<T, D : Dimension>(val point: Point<T>) :
+    DPoint<T, D> {
     override val size: Int get() = point.size
 
     override fun get(index: Int): T = point[index]
@@ -94,7 +95,12 @@ inline class DMatrixContext<T : Any, Ri : Ring<T>>(val context: GenericMatrixCon
 
     inline fun <reified D : Dimension> point(noinline initializer: (Int) -> T): DPoint<T, D> {
         val size = Dimension.dim<D>()
-        return DPoint.coerceUnsafe(context.point(size.toInt(), initializer))
+        return DPoint.coerceUnsafe(
+            context.point(
+                size.toInt(),
+                initializer
+            )
+        )
     }
 
     inline infix fun <reified R1 : Dimension, reified C1 : Dimension, reified C2 : Dimension> DMatrix<T, R1, C1>.dot(
diff --git a/kmath-dimensions/src/jsMain/kotlin/scientifik/kmath/dimensions/dim.kt b/kmath-dimensions/src/jsMain/kotlin/scientifik/kmath/dimensions/dim.kt
new file mode 100644
index 000000000..557234d84
--- /dev/null
+++ b/kmath-dimensions/src/jsMain/kotlin/scientifik/kmath/dimensions/dim.kt
@@ -0,0 +1,5 @@
+package scientifik.kmath.dimensions
+
+actual inline fun <reified D : Dimension> Dimension.Companion.dim(): UInt {
+    TODO("KClass::objectInstance does not work")
+}
\ No newline at end of file
diff --git a/kmath-dimensions/src/jvmMain/kotlin/scientifik/kmath/dimensions/dim.kt b/kmath-dimensions/src/jvmMain/kotlin/scientifik/kmath/dimensions/dim.kt
new file mode 100644
index 000000000..a16dd2266
--- /dev/null
+++ b/kmath-dimensions/src/jvmMain/kotlin/scientifik/kmath/dimensions/dim.kt
@@ -0,0 +1,4 @@
+package scientifik.kmath.dimensions
+
+actual inline fun <reified D : Dimension> Dimension.Companion.dim(): UInt =
+    D::class.objectInstance?.dim ?: error("Dimension object must be a singleton")
\ No newline at end of file
diff --git a/kmath-dimensions/src/commonTest/kotlin/scientifik/kmath/dimensions/DMatrixContextTest.kt b/kmath-dimensions/src/jvmTest/kotlin/scientifik/kmath/dimensions/DMatrixContextTest.kt
similarity index 100%
rename from kmath-dimensions/src/commonTest/kotlin/scientifik/kmath/dimensions/DMatrixContextTest.kt
rename to kmath-dimensions/src/jvmTest/kotlin/scientifik/kmath/dimensions/DMatrixContextTest.kt