Simplified unit testing