another simpler fix for the differential rotation testing