Yes, you should apply wax after the paint correction is completed. The paint correction will remove the scratches and restore the paint's gloss, but also remove the protective layer. This step will leave the paint vulnerable, so you need extra protection. The best thing to do is apply wax or ceramic coating to protect the paint from dirt, acid rain, and UV rays.