Tiny cleanup

Please, check this