Last min bug fixes