code quality

I'm tired today :/