More bug fixes v2