Very small changes I forgot in the last commit..