fixed small bug by adding import