Changing test code to production