Small couple of changes forgot to save