Updating tests for prod env