do support - To use ‘to’ or not is the question ;)

I believe the grammatically correct way is to omit the to before the verb (find in this case). Can someone confirm?

What you have to do instead is find a way to redefine the problem ...

I ask this because I have also seen constructions like

What you have to do instead is to find a way to redefine the problem ...

Are both correct?