Could easily be wheel alignment. They should have performed a wheel alignment on it when they replaced the unit. I would take it back to them and have them fix it. What most likely happened is that they just slammed the new unit in and let it go.
Even if it is not the alignment and something is loose/wrong, it sounds like their work is to blame.