From 49e247dd0d9e6da6c686fb87071fcf7868a770a0 Mon Sep 17 00:00:00 2001
From: Gabriel Augendre <gabriel@augendre.info>
Date: Sun, 9 Jun 2019 17:11:47 +0200
Subject: [PATCH] Allow setting unit on machine creation page

---
 gym/views.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gym/views.py b/gym/views.py
index 7474c6e..ee7a5d6 100644
--- a/gym/views.py
+++ b/gym/views.py
@@ -133,7 +133,7 @@ class EquipmentDetailView(LoginRequiredMixin, QuickActionsMixin, generic.DetailV
 
 class EquipmentCreateView(LoginRequiredMixin, QuickActionsMixin, generic.CreateView):
     model = Equipment
-    fields = ['room', 'name', 'default_work_form', 'default_repetition_number']
+    fields = ['room', 'name', 'default_work_form', 'unit', 'default_repetition_number']
     template_name = 'gym/equipment_edit.html'
     room = None