Comply to Callee structure

1 files changed, 7 insertions, 2 deletions

diff --git a/ethchain/contract.go b/ethchain/contract.go
index 21ac828fe..10c5e2df6 100644
--- a/ethchain/contract.go
+++ b/ethchain/contract.go
@@ -43,12 +43,17 @@ func (c *Contract) State() *State {
 	return c.state
 }
 
-func (c *Contract) GetMem(num int) *ethutil.Value {
-	nb := ethutil.BigToBytes(big.NewInt(int64(num)), 256)
+func (c *Contract) GetMem(num int64) *ethutil.Value {
+	nb := ethutil.BigToBytes(big.NewInt(num), 256)
 
 	return c.Addr(nb)
 }
 
+// Return the gas back to the origin. Used by the Virtual machine or Closures
+func (c *Contract) ReturnGas(val *big.Int, state *State) {
+	c.Amount.Add(c.Amount, val)
+}
+
 func MakeContract(tx *Transaction, state *State) *Contract {
 	// Create contract if there's no recipient
 	if tx.IsContract() {