fixing the warning suggested