Fixed code warning