Adding hook to restart puma (test)