Missed some bits in the tightening

Signed-off-by: Gawain Lynch <gawain.lynch@gmail.com>