Fixing some issue (pylint)