Fix for Code Climate