A small refactor was made.