Fix misformatted code