We shouldn't do this kind of logic in the setup file, especially that it's not cached. I will work on it on a separate branch