Updated tests to do actually testing