bug fixes

Signed-off-by: Nick Jonas <njonas45@gmail.com>