Update swagger so it works