Fix missing argument and tidy up some whitespace.