updated to pass newer test