quick fix got rid of small bug