Fix bugs

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