Add fixes for puma deployment