add fix for testing stage