Fixed minor bug.

Signed-off-by: Dave Bush <davebush@gmail.com>