bug fixes!