trying to add tests but it's not working