bug fixes fixes