Make sure you do defensive puzzles as well. I'm fairly decent at spotting a tactic to win material a few moves ahead, but terrible at seeing one coming from my opponent early enough to avert it. I'm not sure what lichess has, but definitely look for puzzles where you have to avoid a knight fork or some pin or incoming checkmate.