Fixed puma issue