small correct
Revert "small correct"

This reverts commit 43d1c0487af620cfaf7a41f1f02d1090e40a351f.
Small fixes