Compatibility with the current code

Signed-off-by: Siddhartha Gupta <makerzzz.zzzz@gmail.com>