Fixing first test. Now passes