Applied an IDEA suggestion