Remove tests for production