fix things broken in the revert

Signed-off-by: Chris Granger <ibdknox@gmail.com>