bugfix, +tests