Added missing tests... whoops