Fix couple tests