Fix copy and paste issue in last commit