bug fixes

yay bugs