fixed error in setup code