Get the type correct

Signed-off-by: Tom Haynes <loghyr@primarydata.com>