Use containsKey to fix silly NPE