Code climate fix