diff --git a/third_max.rb b/third_max.rb
index d959034..067891e 100644
--- a/third_max.rb
+++ b/third_max.rb
@@ -1,3 +1,8 @@
 def third_max(nums)
-    return nums.uniq.sort[-3] 
+    sortnums = nums.uniq.sort
+    if sortnums.length < 3
+        return sortnums.last
+    else
+        return sortnums[-3]
+    end    
 end
\ No newline at end of file