Fix some bugs.

Signed-off-by: Tyler O'Meara <Tyler@TylerOMeara.com>