Minor cleanup

Signed-off-by: Cody Massin <keyosk@gmail.com>