Test update

small bug fix