The grammar is correct in both. But, we are more likely to use was looking for, particularly when you find something you need.

That's exactly what I was looking for.