Fixed bug in quantum setup