Minor bug fixes

Signed-off-by: Cory Feitelson <cfeitelson@roovy.com>