fix up code

Signed-off-by: Dave Moos <dmoos@datarecognition.com>