fix whitespace and re-add 2 lines accidentally deleted