Minor clean up
Fix wrong copy-pasting