Fixing warning caused by last commit