fixed more stuff

Signed-off-by: Zach Klabunde <zbunde@gmail.com>