Remove it from stub as well