Fix (?) mock errors