wip
removing a small bug