They are both OK. Personally I prefer to include "that", especially in writing, but in this case the "that ... that" repetition is noticeable (not glaringly so, but possibly just enough to consider how to avoid it).
New words, one handy idiom, and a 2-minute quiz — delivered to your inbox to keep your streak alive.