fixed a bug in the setup