fix steps typing in mock extension refactoring