nitvm: Bug fixes