Added ComprehensionBlock to our tests