Fix error in setup