hotfixed a small bug