Fixed error in setup