minor fix in setup