Slight error fixes