minor bug fixes v2