Add missing bits

Signed-off-by: Andrew 'Necromant' Andrianov <andrew@ncrmnt.org>