bad extend class after refactor