Bug fixes v2