fixing setup
fixing reg bug