Исправлено на семинаре.

Set использовать не нужно было, нужно было использовать Map