We need to have tests for the prototype version as well (and sharing things as much as possible to ensure they behave the same)