Fix lshift tests