diff --git a/lib/new-admin/graphql/resolvers/customer.resolver.js b/lib/new-admin/graphql/resolvers/customer.resolver.js
index 9ebf636f..059cbe4f 100644
--- a/lib/new-admin/graphql/resolvers/customer.resolver.js
+++ b/lib/new-admin/graphql/resolvers/customer.resolver.js
@@ -14,6 +14,7 @@ const resolvers = {
   },
   Mutation: {
     setCustomer: (root, { customerId, customerInput }, context, info) => {
+      // To be replaced by function that fetchs the token
       const token = !!context.req.cookies.lamassu_sid && context.req.session.user.id
       if (customerId === anonymous.uuid) return customers.getCustomerById(customerId)
       return customers.updateCustomer(customerId, customerInput, token)
@@ -22,11 +23,13 @@ const resolvers = {
     saveCustomField: (...[, { customerId, fieldId, newValue }]) => customers.saveCustomField(customerId, fieldId, newValue),
     removeCustomField: (...[, [ { customerId, fieldId } ]]) => customers.removeCustomField(customerId, fieldId),
     editCustomer: async (root, { customerId, customerEdit }, context) => {
+      // To be replaced by function that fetchs the token
       const token = !!context.req.cookies.lid && context.req.session.user.id
       const editedData = await customerEdit
       return customers.edit(customerId, editedData, token)
     },
     replacePhoto: async (root, { customerId, photoType, newPhoto }, context) => {
+      // To be replaced by function that fetchs the token
       const token = !!context.req.cookies.lid && context.req.session.user.id
       const photo = await newPhoto
       if (!photo) return customers.getCustomerById(customerId)
diff --git a/new-lamassu-admin/src/components/buttons/FeatureButton.js b/new-lamassu-admin/src/components/buttons/FeatureButton.js
index 86a06c5e..3fe97cd7 100644
--- a/new-lamassu-admin/src/components/buttons/FeatureButton.js
+++ b/new-lamassu-admin/src/components/buttons/FeatureButton.js
@@ -32,8 +32,8 @@ const styles = {
 const useStyles = makeStyles(styles)
 
 const FeatureButton = memo(
-  ({ className, Icon, InverseIcon, children, active, ...props }) => {
-    const classes = useStyles({ active })
+  ({ className, Icon, InverseIcon, children, ...props }) => {
+    const classes = useStyles()
 
     const classNames = {
       [classes.featureButton]: true,