diff --git a/functions/matrix.go b/functions/matrix.go
index c844724..9cc6bf6 100644
--- a/functions/matrix.go
+++ b/functions/matrix.go
@@ -1,7 +1,10 @@
 package kubeless
 
-import "fmt"
+import (
+	"fmt"
 
+	"github.com/kubeless/kubeless/pkg/functions"
+)
 func main() {
 	//Defining 2D matrices
 	m1 := [3][3]int{