small code fix in setup