Fix transaction mocking