Allow both Stage* and Node* for annotation support

See #88.