Можно убрать containsKey и проверять результат get на null.