Minor change in the setup