Two lines missed from the previous patch.
Signed-off-by: NSteven Whitehouse <swhiteho@redhat.com>