Removed debug code

The code might be crashing Spark Core