Small cleanup

Removed some warnings