Fixed silly mistakes when adding tests