Cleanup some code.

Signed-off-by: TomorrowLandAce <joaolrpaulo@gmail.com>