Style changes to test case as per @krasnouhkov's request.