Yay bug fixes :)