changes to for natspec

2 files changed, 3 insertions, 1 deletions

diff --git a/index.js b/index.js
index e0382a8a4..0c5bf3d44 100644
--- a/index.js
+++ b/index.js
@@ -7,5 +7,6 @@ web3.providers.HttpRpcProvider = require('./lib/httprpc');
 web3.providers.QtProvider = require('./lib/qt');
 web3.providers.AutoProvider = require('./lib/autoprovider');
 web3.eth.contract = require('./lib/contract');
+web3.abi = require('./lib/abi');
 
 module.exports = web3;
diff --git a/lib/contract.js b/lib/contract.js
index 67fb407ad..abd8e5bdf 100644
--- a/lib/contract.js
+++ b/lib/contract.js
@@ -79,7 +79,8 @@ var contract = function (address, desc) {
                     extra.to = address;
                     return abi.methodSignature(desc, method.name).then(function (signature) {
                         extra.data = signature.slice(0, 2 + ETH_METHOD_SIGNATURE_LENGTH * 2) + parsed;
-                        web3._currentAbi = desc;
+                        web3._currentContractAbi = desc;
+                        web3._currentContractAddress = address;
                         return web3.eth.transact(extra).then(onSuccess);
                     });
                 }