fixed mock and test