writing tests (also fixed build :))