Added missing tests to reflect changes