Also set stuck class on wrapper element