Added missing setup part