forgot adding some tests