modified tests to pass for new feature