Fixed weird code