fixed code to reflect the example in setup file