remove generic
add mock