Yay bug fixes