Fix missing package in tests

Signed-off-by: John Tuley <jtuley@pivotal.io>