Minor update to fix checkstyle error