diff --git a/controllers/userController.go b/controllers/userController.go
index 8d1bf19..89059c4 100644
--- a/controllers/userController.go
+++ b/controllers/userController.go
@@ -13,6 +13,7 @@ import (
 	"golang-test/types"
 	"golang-test/validators"
 	"golang.org/x/crypto/bcrypt"
+	"strings"
 )
 
 //func tokenGenerator() string {
@@ -113,7 +114,9 @@ func GetUserByToken(token string) (database.User, error) {
 	return user, err
 }
 
-func EditUser(c *gin.Context, token string, user database.User) {
+func EditUser(c *gin.Context, user database.User) {
+	token := strings.Split(c.Request.Header["Bearer"][0], " ")[1]
+
 	db := database.Connector()
 	validate := validators.Validate
 	response := message.Response{Status: 200}
@@ -161,7 +164,8 @@ func EditUser(c *gin.Context, token string, user database.User) {
 	message.SendResponse(c, response)
 }
 
-func DeleteUser(c *gin.Context, token string) {
+func DeleteUser(c *gin.Context) {
+	token := strings.Split(c.Request.Header["Bearer"][0], " ")[1]
 	db := database.Connector()
 	u, err := GetUserByToken(token)
 	response := message.Response{Status: 200}