Minor change to fix some warnings