bug fix in setup