Fix puma tasks