Fixed bug in special case code where it was getting called for first map phase