From 00f323d132c08c4eb60f32c9e50e37ccf2118539 Mon Sep 17 00:00:00 2001
From: Francisco Giordano <frangio.1@gmail.com>
Date: Mon, 28 Aug 2017 17:05:15 -0300
Subject: [PATCH] add shortcut for vestedAmount after vesting end

---
 contracts/token/TokenVesting.sol | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/contracts/token/TokenVesting.sol b/contracts/token/TokenVesting.sol
index a445c644a..6eb2e17d2 100644
--- a/contracts/token/TokenVesting.sol
+++ b/contracts/token/TokenVesting.sol
@@ -70,6 +70,8 @@ contract TokenVesting is Ownable {
   function vestedAmount(ERC20Basic token) constant returns (uint256) {
     if (now < cliff) {
       return 0;
+    } else if (now >= end) {
+      return token.balanceOf(this);
     } else {
       uint256 currentBalance = token.balanceOf(this);
       uint256 totalBalance = currentBalance.add(released[token]);