fix stupid refactor bug