Fixed so code should run in VM now