Fixed code quality problem