Add tests to prod version