Fixing JSON bug