Fixed mock to abide by interface