Fix a bug in State and AwareVM