BUG: Corrected bug in setup