Update tests for production