fix count problem in controller
fix up things :)
better