From fa897a0d1a9db9faf62a694caf844892527cf43d Mon Sep 17 00:00:00 2001
From: Pol Quintana <pol.quintana1@gmail.com>
Date: Thu, 16 Jun 2016 14:24:44 +0200
Subject: [PATCH] Avoid force unwraps that could crash on tests - Fixes #87

---
 ActiveLabelTests/ActiveTypeTests.swift | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/ActiveLabelTests/ActiveTypeTests.swift b/ActiveLabelTests/ActiveTypeTests.swift
index 508cf743..e97fdc81 100644
--- a/ActiveLabelTests/ActiveTypeTests.swift
+++ b/ActiveLabelTests/ActiveTypeTests.swift
@@ -29,8 +29,8 @@ class ActiveTypeTests: XCTestCase {
         return label.activeElements.flatMap({$0.1.flatMap({$0.element})})
     }
     
-    var currentElementString: String {
-        let currentElement = activeElements.first!
+    var currentElementString: String? {
+        guard let currentElement = activeElements.first else { return nil }
         switch currentElement {
         case .Mention(let mention):
             return mention
@@ -43,8 +43,8 @@ class ActiveTypeTests: XCTestCase {
         }
     }
     
-    var currentElementType: ActiveType {
-        let currentElement = activeElements.first!
+    var currentElementType: ActiveType? {
+        guard let currentElement = activeElements.first else { return nil }
         switch currentElement {
         case .Mention:
             return .Mention