diff --git a/app/controllers/activeSprintsController.js b/app/controllers/activeSprintsController.js
index f10a389..d32855c 100644
--- a/app/controllers/activeSprintsController.js
+++ b/app/controllers/activeSprintsController.js
@@ -57,6 +57,16 @@ app.controller('ActiveSprintsController', function ($controller, $sce, $scope, A
         return $scope.activeSprints.length > 0 ? $scope.activeSprints[sessionStorage.selected] : undefined;
     };
 
+    $scope.getPanelClass = function (type) {
+        var panelClass = 'panel-default';
+        if (type === 'Feature') {
+            panelClass = 'panel-primary';
+        } else if (type === 'Defect') {
+            panelClass = 'panel-danger';
+        }
+        return panelClass;
+    };
+
     ActiveSprintsService.updated.then(null, null, function () {
         if ($scope.activeSprints.length > 0) {
             while (sessionStorage.selected >= $scope.activeSprints.length) {
diff --git a/app/views/home.html b/app/views/home.html
index 7dc3bfa..6004c56 100644
--- a/app/views/home.html
+++ b/app/views/home.html
@@ -50,7 +50,7 @@
                   <div class="panel-heading">{{status.identifier}}</div>
                   <div class="panel-body">
 
-                    <div ng-repeat="card in getSelectedSprint().cards | filter:{'status': status.identifier}" class="panel kanban-card" ng-class="{'panel-primary': card.type === 'Feature', 'panel-danger': card.type === 'Defect'}">
+                    <div ng-repeat="card in getSelectedSprint().cards | filter:{'status': status.identifier}" class="panel kanban-card" ng-class="getPannelClass(card.type)">
                       <div class="panel-heading">
                         <span>{{card.number}}</span>
                         <span ng-repeat="assignee in card.assignees">
diff --git a/karma.conf.js b/karma.conf.js
index 6d19828..7409e95 100644
--- a/karma.conf.js
+++ b/karma.conf.js
@@ -121,7 +121,7 @@ module.exports = function (config) {
 
         frameworks: ['jasmine'],
 
-        browsers: ['Chrome', 'Firefox'],
+        browsers: ['ChromeHeadless', 'Firefox'],
 
         plugins: [
             'karma-chrome-launcher',
diff --git a/tests/unit/controllers/activeSprintsControlerTest.js b/tests/unit/controllers/activeSprintsControlerTest.js
index c118655..61842b9 100644
--- a/tests/unit/controllers/activeSprintsControlerTest.js
+++ b/tests/unit/controllers/activeSprintsControlerTest.js
@@ -53,6 +53,10 @@ describe('controller: ActiveSprintsController', function () {
             expect(scope.getSelectedSprint).toBeDefined();
             expect(typeof scope.getSelectedSprint).toEqual('function');
         });
+        it('getPanelClass should be defined', function () {
+            expect(scope.getPanelClass).toBeDefined();
+            expect(typeof scope.getPanelClass).toEqual('function');
+        });
     });
 
     describe('Do the scope methods work as expected', function () {
@@ -95,6 +99,15 @@ describe('controller: ActiveSprintsController', function () {
             var selectedSprint = scope.getSelectedSprint();
             expect(selectedSprint).toEqual(scope.activeSprints[0]);
         });
+
+        it('getPanelClass should return correct value', function() {
+            var featureClass = scope.getPanelClass('Feature');
+            var defectClass = scope.getPanelClass('Defect');
+            var otherClass = scope.getPanelClass('anything else');
+            expect(featureClass).toEqual('panel-primary');
+            expect(defectClass).toEqual('panel-danger');
+            expect(otherClass).toEqual('panel-default');
+        });
     });
 
 });
\ No newline at end of file