Fix checkstyle
Issue 401