Simplified test code