Woops, forgot to fix test expect call.