Fixing the first test