Add missing code back in so that product generation works