bug fixes yo!!