Are you sure you are not mixing up img and images?
Additionally to the unsqueezed dimension img also has suddently a batch size of 4, which wasn’t there previously.
Could you print images.shape before unsqueezing it?