Minor changes

Minor changes to stop the warning message