Minor fixes

FIXED LIKE A LORD