Fix bug

copy and paste can get a bit clutsy.