Fix expect call